Update config.py

config.py

@@ -5,23 +5,29 @@ class Config:
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
     DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 
-    # Base Model & LoRA
+    # Base Model & LoRA (from primerz/pixagram)
     REPO_ID = "primerz/pixagram"
     CHECKPOINT_FILENAME = "horizon.safetensors"
     LORA_FILENAME = "retroart.safetensors"
     
-    # Trigger Words
+    # Trigger Words for the LoRA
     STYLE_TRIGGER = "p1x3l4rt, pixel art"
     
     # InstantID Assets
     INSTANTID_REPO = "InstantX/InstantID"
-    FACE_ADAPTER_REPO = "InstantX/InstantID"
-    FACE_ADAPTER_FILENAME = "ip-adapter.bin" # Usually auto-loaded by diffusers helper
+    CN_INSTANTID_REPO = "InstantX/InstantID" # Contains the ControlNet part
     
     # ControlNet Repos
     CN_ZOE_REPO = "diffusers/controlnet-zoe-depth-sdxl-1.0"
-    CN_LINEART_REPO = "xinsir/controlnet-lineart-sdxl-1.0" # High quality lineart
-    CN_INSTANTID_REPO = "InstantX/InstantID" # Contains the ControlNet part
+    CN_LINEART_REPO = "xinsir/controlnet-lineart-sdxl-1.0"
+    
+    # Preprocessor (Annotator) Repo
+    ANNOTATOR_REPO = "lllyasviel/Annotators"
+    
+    # Captioning Model
+    CAPTIONER_REPO = "Salesforce/blip-image-captioning-base"
     
-    # Vision Models (for InsightFace)
-    ANTELOPEV2_PATH = "./models/antelopev2" # You might need to download this manually in utils
+    # InsightFace Model (HF Hub mirror)
+    ANTELOPEV2_REPO = "casperhansen/antelopev2"
+    ANTELOPEV2_ROOT = "./models" # Parent folder
+    ANTELOPEV2_NAME = "antelopev2"