Update app_zero.py

app_zero.py

@@ -1,7 +1,5 @@
 import sys
-import os
 import types
-import random
 import datetime
 from pathlib import Path
 
@@ -96,7 +94,6 @@ def patch_file(path_str: str, replacements: list[tuple[str, str]]) -> None:
         return
 
     original = text
-
     for old, new in replacements:
         text = text.replace(old, new)
 
@@ -122,7 +119,7 @@ def patch_pasd_for_diffusers() -> None:
         ],
     )
 
-    # 2) PositionNet -> GLIGENTextBoundingboxProjection alias
+    # 2) PositionNet was replaced by GLIGENTextBoundingboxProjection
     patch_file(
         "./PASD/models/pasd/unet_2d_condition.py",
         [
@@ -153,21 +150,38 @@ def patch_pasd_for_diffusers() -> None:
         ],
     )
 
-    # 4) normalize ControlNet loader import for diffusers 0.27.x
+    # 4) robust ControlNet mixin import fallback
+    controlnet_import_fallback = '''try:
+    from diffusers.loaders import FromOriginalControlNetMixin as FromOriginalControlnetMixin
+except ImportError:
+    try:
+        from diffusers.loaders import FromOriginalControlnetMixin
+    except ImportError:
+        class FromOriginalControlnetMixin:
+            pass'''
+
     patch_file(
         "./PASD/models/pasd/controlnet.py",
         [
             (
-                "from diffusers.loaders.single_file_model import FromOriginalModelMixin as FromOriginalControlnetMixin",
                 "from diffusers.loaders import FromOriginalControlnetMixin",
+                controlnet_import_fallback,
+            ),
+            (
+                "from diffusers.loaders import FromOriginalControlNetMixin",
+                controlnet_import_fallback,
+            ),
+            (
+                "from diffusers.loaders.single_file_model import FromOriginalModelMixin as FromOriginalControlnetMixin",
+                controlnet_import_fallback,
             ),
             (
                 "from diffusers.loaders.single_file_model import FromOriginalModelMixin as FromOriginalControlNetMixin",
-                "from diffusers.loaders import FromOriginalControlnetMixin",
+                controlnet_import_fallback,
             ),
             (
-                "from diffusers.loaders import FromOriginalControlNetMixin",
-                "from diffusers.loaders import FromOriginalControlnetMixin",
+                "FromOriginalControlNetMixin",
+                "FromOriginalControlnetMixin",
             ),
         ],
     )