feat 取消flux 的污点判断

fix
2024-12-01 19:03:55 +08:00
parent b5b52f5fbe
commit 6a0c581e5d
1 changed files with 1 additions and 1 deletions
--- 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":
+            if self.category == "sketch" and self.version == "fast":
                # 色阶调整
                cutoff = 1
                levels_img = autoLevels(image_result, cutoff)