From 0f17d8b6544288b3f8a3269ca75f96ea04b4e5bc Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Sun, 1 Dec 2024 19:52:33 +0800 Subject: [PATCH] =?UTF-8?q?feat=20=20flux=20=E5=8F=96=E6=B6=88=E6=B1=A1?= =?UTF-8?q?=E7=82=B9=E6=A3=80=E6=B5=8B=20=E5=A2=9E=E5=8A=A0=E7=B1=BB?= =?UTF-8?q?=E5=88=AB=E5=88=A4=E6=96=AD=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/generate_image/service_generate_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/service/generate_image/service_generate_image.py b/app/service/generate_image/service_generate_image.py index 2ba7862..86912f8 100644 --- a/app/service/generate_image/service_generate_image.py +++ b/app/service/generate_image/service_generate_image.py @@ -91,7 +91,7 @@ class GenerateImage: image = result.as_numpy("generated_image") image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_RGB2BGR) is_smudge = True - if self.category == "sketch" and self.version == "fast": + if self.category == "sketch": if self.version == "fast": # 色阶调整 cutoff = 1