Update config.py

config.py

@@ -1,21 +1,23 @@
-import torch
 
 class Config:
     # Hardware
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
     DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 
-    # Base Model & LoRA (from primerz/pixagram)
-    REPO_ID = "primerz/pixagram"
-    CHECKPOINT_FILENAME = "horizon.safetensors"
-    LORA_FILENAME = "retroart.safetensors"
-    LORA_STRENGTH = 1.15 # Fixed strength for fusion
+    # --- 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 = "lucasart.safetensors"
+    LORA_STRENGTH = 1.0 # TCD works well with 1.0
     
     # Trigger Words for the LoRA
-    STYLE_TRIGGER = "p1x3l4rt, pixel art"
+    STYLE_TRIGGER = "lucasarts style"
     
-    # Default Negative Prompt
-    DEFAULT_NEGATIVE_PROMPT = "Ugly, artifacts, blurry, disformed, photo-realistic, photo, photography, realistic, low-quality, pixelart, text."
+    # Default Negative Prompt (Updated for general use)
+    DEFAULT_NEGATIVE_PROMPT = "cartoon, 3d, (disfigured), (bad art), (deformed), (poorly drawn), (extra limbs), strange colors, blurry, boring, sketch, lacklustre, repetitive, cropped"
+    # --- END UPDATED ---
 
     # InstantID Assets
     INSTANTID_REPO = "InstantX/InstantID"
@@ -35,9 +37,9 @@ class Config:
     ANTELOPEV2_ROOT = "." # Parent folder
     ANTELOPEV2_NAME = "antelopev2"
     
-    # Gradio Parameters
-    CGF_SCALE = 1.5
-    STEPS_NUMBER = 14
+    # Gradio Parameters (CFG_SCALE is no longer used by generator)
+    CGF_SCALE = 0.0 # TCD default
+    STEPS_NUMBER = 8
     IMG_STRENGTH = 0.8
     DEPTH_STRENGTH = 0.8
     EDGE_STRENGTH = 0.8