Update config.py

config.py

@@ -5,30 +5,28 @@ class Config:
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
     DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 
-    # Base Model & Style LoRA
+    # --- UPDATED: New Base Model & Style LoRA ---
+    # Assuming these are in the 'primerz/pixagram' repo or a new one.
+    # If they are in a different repo, change REPO_ID.
     REPO_ID = "primerz/pixagram" 
     CHECKPOINT_FILENAME = "reality.safetensors"
     LORA_FILENAME = "retroart.safetensors"
-    LORA_STRENGTH = 1.25
+    LORA_STRENGTH = 1.25 # TCD works well with 1.0
     
     # Trigger Words for the LoRA
     STYLE_TRIGGER = "p1x3l4rt, pixel art"
     
-    # Default Negative Prompt
+    # Default Negative Prompt (Updated for general use)
     DEFAULT_NEGATIVE_PROMPT = "Ugly, artifacts, blurry, disformed, photo-realistic, photo, photography, realistic, low-quality, pixelart, text."
+    # --- END UPDATED ---
 
     # InstantID Assets
     INSTANTID_REPO = "InstantX/InstantID"
     
     # ControlNet Repos
-    CN_ZOE_REPO = "diffusers/controlnet-zoe-depth-sdxl-1.0"
-    CN_CANNY_REPO = "diffusers/controlnet-canny-sdxl-1.0"
+    CN_ZOE_REPO = "diffusers/controlnet-zoE-depth-sdxl-1.0"
     CN_LINEART_REPO = "ShermanG/ControlNet-Standard-Lineart-for-SDXL"
     
-    # Regular IP-Adapter (for style transfer)
-    IP_ADAPTER_REPO = "h94/IP-Adapter"
-    IP_ADAPTER_FILENAME = "ip-adapter_sdxl.bin"
-    
     # Preprocessor (Annotator) Repo
     ANNOTATOR_REPO = "lllyasviel/Annotators"
     
@@ -37,20 +35,13 @@ class Config:
     
     # InsightFace Model (HF Hub mirror)
     ANTELOPEV2_REPO = "DIAMONIK7777/antelopev2"
-    ANTELOPEV2_ROOT = "."
+    ANTELOPEV2_ROOT = "." # Parent folder
     ANTELOPEV2_NAME = "antelopev2"
     
-    # TCD Optimized Defaults
-    CGF_SCALE = 4.0  # Good for TCD + LoRA
-    STEPS_NUMBER = 8
-    IMG_STRENGTH = 0.8
+    # Gradio Parameters
+    # --- DPMSolver++ works well with standard CFG and more steps ---
+    CGF_SCALE = 7.0 # Standard guidance scale
+    STEPS_NUMBER = 20 # DPMSolver++ recommended steps
+    IMG_STRENGTH = 0.85
     DEPTH_STRENGTH = 0.8
-    EDGE_STRENGTH = 0.8
-    
-    # Control Type Options
-    EDGE_TYPES = ["canny", "lineart", "both"]  # New: Support multiple edge detection types
-    DEFAULT_EDGE_TYPE = "canny"
-    
-    # IP-Adapter Settings
-    INSTANTID_SCALE = 0.8  # For face preservation
-    STYLE_IP_SCALE = 0.5   # For style transfer
+    EDGE_STRENGTH = 0.8