Update model.py

model.py

@@ -68,7 +68,7 @@ class ModelHandler:
 
         # 2. Load ControlNets
         # --- MODIFIED: Updated print ---
-        print("Loading ControlNets (InstantID, Zoe, LineArt, Tile)...")
+        print("Loading ControlNets (InstantID, Zoe, LineArt)...")
 
         # Load the InstantID ControlNet from the correct subfolder
         print("Loading InstantID ControlNet from subfolder 'ControlNetModel'...")
@@ -80,17 +80,17 @@ class ModelHandler:
         print("  [OK] Loaded InstantID ControlNet.")
         
         # Load other ControlNets normally
-        # --- MODIFIED: Load Tile CN ---
-        print("Loading Zoe, LineArt, and Tile ControlNets...")
+        # --- MODIFIED: Removed Tile CN ---
+        print("Loading Zoe and LineArt ControlNets...")
         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)
-        cn_tile = ControlNetModel.from_pretrained(Config.CN_TILE_REPO, torch_dtype=Config.DTYPE)
+        # cn_tile = ControlNetModel.from_pretrained(Config.CN_TILE_REPO, torch_dtype=Config.DTYPE) # <-- REMOVED
         # --- END MODIFIED ---
 
         # --- Manually wrap the list of models in a MultiControlNetModel ---
         print("Wrapping ControlNets in MultiControlNetModel...")
-        # --- MODIFIED: Add Tile CN to list ---
-        controlnet_list = [cn_instantid, cn_zoe, cn_lineart, cn_tile]
+        # --- MODIFIED: Removed Tile CN from list ---
+        controlnet_list = [cn_instantid, cn_zoe, cn_lineart]
         controlnet = MultiControlNetModel(controlnet_list)
         # --- End wrapping ---
 
@@ -154,7 +154,7 @@ class ModelHandler:
         print("  [OK] LoRA fused.")
         
         # 6. Load Preprocessors
-        # --- MODIFIED: Removed ColorDetector ---
+        # --- MODIFIED: Removed Tile ---
         print("Loading Preprocessors (LeReS, LineArtAnime)...")
         self.leres_detector = LeresDetector.from_pretrained(Config.ANNOTATOR_REPO)
         self.lineart_anime_detector = LineartAnimeDetector.from_pretrained(Config.ANNOTATOR_REPO)