From 9213296477481159b421aed5d5e68ece5e501d3c Mon Sep 17 00:00:00 2001
From: 10join <140466896+10join@users.noreply.github.com>
Date: Tue, 25 Jul 2023 22:07:17 +0900
Subject: [PATCH] Update txt2img.py

x_checked_image = x_samples_ddim
---
 scripts/txt2img.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/txt2img.py b/scripts/txt2img.py
index bc3864043..a4d4255c2 100644
--- a/scripts/txt2img.py
+++ b/scripts/txt2img.py
@@ -314,8 +314,8 @@ 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)
 
                         if not opt.skip_save: