From 239ed0fd0247807df4f62e5dc22dfe00f6e14db3 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Fri, 19 Aug 2022 17:09:41 +0000
Subject: [PATCH] fix more
---
scripts/txt2img.py | 7 +------
1 file changed, 1 insertion(+), 6 deletions(-)
diff --git a/scripts/txt2img.py b/scripts/txt2img.py
index add30f5..9613689 100644
--- a/scripts/txt2img.py
+++ b/scripts/txt2img.py
@@ -266,7 +266,7 @@ def main():
x_samples_ddim = model.decode_first_stage(samples_ddim)
x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
- x_samples_ddim = x_samples_ddim.cpu().permute(0, 2, 3, 1).numpy()
+ x_samples_ddim = x_samples_ddim.cpu().permute(0, 2, 3, 1)
x_image = x_samples_ddim
safety_checker_input = safety_feature_extractor(numpy_to_pil(x_image), return_tensors="pt")
@@ -295,11 +295,6 @@ def main():
Image.fromarray(grid.astype(np.uint8)).save(os.path.join(outpath, f'grid-{grid_count:04}.png'))
grid_count += 1
- image = x_samples_ddim.cpu().permute(0, 2, 3, 1).numpy()
-
- # run safety checker
- safety_checker_input = pipe.feature_extractor(numpy_to_pil(image), return_tensors="pt")
- image, has_nsfw_concept = pipe.safety_checker(images=image, clip_input=safety_checker_input.pixel_values)
toc = time.time()
print(f"Your samples are ready and waiting for you here: \n{outpath} \n"
--
GitLab