Update model.py

model.py

@@ -79,10 +79,31 @@ class ModelHandler:
         )
         print("  [OK] Loaded InstantID ControlNet.")
         
-        # Load other ControlNets normally
         print("Loading Depth and Edge ControlNets...")
-        cn_depth = Processor(Config.DEPTH)
-        cn_edge = Processor(Config.EDGE)
+
+        # --- FIX: Load the actual ControlNetModels, not the 'controlnet-aux' Processors ---
+        # The 'Processor' class is for pre-processing (creating the maps), 
+        # not for the model itself. Your Config.DEPTH ("lesres++") is an ID
+        # for the pre-processor, which is correctly used at the end of this file.
+        
+        # Define the repo IDs for the standard SDXL ControlNet models
+        DEPTH_MODEL_REPO = "diffusers/controlnet-depth-sdxl-1.0"
+        EDGE_MODEL_REPO = "diffusers/controlnet-lineart-sdxl-1.0" # Matches "LineArt" in your print statements
+        
+        cn_depth = ControlNetModel.from_pretrained(
+            DEPTH_MODEL_REPO,
+            torch_dtype=Config.DTYPE
+        )
+        print(f"  [OK] Loaded Depth ControlNet ({DEPTH_MODEL_REPO}).")
+        
+        cn_edge = ControlNetModel.from_pretrained(
+            EDGE_MODEL_REPO,
+            torch_dtype=Config.DTYPE
+        )
+        print(f"  [OK] Loaded Edge/LineArt ControlNet ({EDGE_MODEL_REPO}).")
+        
+        # --- End fix ---
+
         # --- Manually wrap the list of models in a MultiControlNetModel ---
         print("Wrapping ControlNets in MultiControlNetModel...")
         controlnet_list = [cn_instantid, cn_depth, cn_edge]
@@ -153,6 +174,8 @@ class ModelHandler:
         #     print(f"  [WARNING] torch.compile failed: {e}. Running without compilation.")
         
         # 6. Load Preprocessors
+        # This is the CORRECT place to use the 'Processor' class
+        # Config.DEPTH ("lesres++") and Config.EDGE are the annotator IDs
         print("Loading Preprocessors (Zoe, LineArt)...")
         self.depth_detector = Processor(Config.DEPTH)
         self.edge_detector = Processor(Config.EDGE)