Update model.py

model.py

@@ -9,8 +9,8 @@ from diffusers import (
     LCMScheduler
 )
 # --- START FIX ---
-# Import the MultiControlNetModel wrapper
-from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel
+# Import MultiControlNetModel from its new, correct location
+from diffusers.models.controlnets.multicontrolnet import MultiControlNetModel
 # --- END FIX ---
 
 # Import the custom pipeline from your local file
@@ -87,12 +87,11 @@ class ModelHandler:
         cn_zoe = ControlNetModel.from_pretrained(Config.CN_ZOE_REPO, torch_dtype=Config.DTYPE)
         cn_lineart = ControlNetModel.from_pretrained(Config.CN_LINEART_REPO, torch_dtype=Config.DTYPE)
 
-        # --- START FIX for AssertionError ---
-        # Manually wrap the list of models in a MultiControlNetModel
+        # --- Manually wrap the list of models in a MultiControlNetModel ---
         print("Wrapping ControlNets in MultiControlNetModel...")
         controlnet_list = [cn_instantid, cn_zoe, cn_lineart]
         controlnet = MultiControlNetModel(controlnet_list)
-        # --- END FIX ---
+        # --- End wrapping ---
 
         # 3. Load SDXL Pipeline
         print(f"Loading SDXL Pipeline ({Config.CHECKPOINT_FILENAME})...")
@@ -112,7 +111,7 @@ class ModelHandler:
         print(f"Loading pipeline from local file: {checkpoint_local_path}")
         self.pipeline = StableDiffusionXLInstantIDImg2ImgPipeline.from_single_file(
             checkpoint_local_path, # Pass the local path
-            controlnet=controlnet, # <-- Pass the single, wrapped object
+            controlnet=controlnet, # Pass the single, wrapped object
             torch_dtype=Config.DTYPE,
             use_safetensors=True
         )