From 14e3c997ef6afc48d971d89b3964206ffdd33895 Mon Sep 17 00:00:00 2001 From: 10join <140466896+10join@users.noreply.github.com> Date: Tue, 25 Jul 2023 21:57:30 +0900 Subject: [PATCH] Update txt2img.py --- scripts/txt2img.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/txt2img.py b/scripts/txt2img.py index bc3864043..043bdb82d 100644 --- a/scripts/txt2img.py +++ b/scripts/txt2img.py @@ -314,7 +314,7 @@ def main(): 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_checked_image, has_nsfw_concept = check_safety(x_samples_ddim) + x_checked_image = x_samples_ddim x_checked_image_torch = torch.from_numpy(x_checked_image).permute(0, 3, 1, 2)