fix: webui < 1.6.0 import error

Bing-su · Mar 25, 2024 · c083197 · c083197
1 parent 0a07045
commit c083197
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
@@ -50,13 +50,20 @@
 from modules.processing import (
     Processed,
     StableDiffusionProcessingImg2Img,
-    create_binary_mask,
     create_infotext,
     process_images,
 )
 from modules.sd_samplers import all_samplers
 from modules.shared import cmd_opts, opts, state
 
+try:
+    from modules.processing import create_binary_mask
+except ImportError:
+
+    def create_binary_mask(image: Image.Image):
+        return image.convert("L")
+
+
 if TYPE_CHECKING:
     from fastapi import FastAPI