Update config.py

config.py

@@ -1,5 +1,6 @@
 """
 Configuration file for Pixagram AI Pixel Art Generator
+UPDATED VERSION with InstantID configuration
 """
 import os
 import torch
@@ -22,11 +23,20 @@ MODEL_FILES = {
 # LORA configuration
 TRIGGER_WORD = "p1x3l4rt, pixel art"
 
-# Face detection configuration
+# InstantID configuration (NEW)
+INSTANTID_CONFIG = {
+    "repo": "InstantX/InstantID",
+    "controlnet_subfolder": "ControlNetModel",
+    "ip_adapter_file": "ip-adapter.bin",
+    "default_ip_scale": 0.8
+}
+
+# Face detection configuration (UPDATED)
 FACE_DETECTION_CONFIG = {
     "model_name": "antelopev2",
     "det_size": (640, 640),
-    "ctx_id": 0
+    "ctx_id": 0,
+    "root": "./models/insightface"
 }
 
 # Recommended resolutions
@@ -187,8 +197,44 @@ CLIP_SKIP = 2
 # Identity boost multiplier
 IDENTITY_BOOST_MULTIPLIER = 1.15
 
-print(f"[CONFIG] Loaded configuration")
+
+def validate_presets():
+    """Validate all preset configurations"""
+    errors = []
+    
+    valid_ranges = {
+        'strength': (0.0, 1.0),
+        'guidance_scale': (0.5, 2.0),
+        'identity_preservation': (0.0, 2.5),
+        'lora_scale': (0.0, 2.0),
+        'depth_control_scale': (0.0, 1.5),
+        'identity_control_scale': (0.0, 1.5)
+    }
+    
+    for preset_name, preset in PRESETS.items():
+        for param, (min_val, max_val) in valid_ranges.items():
+            if param in preset:
+                value = preset[param]
+                if not (min_val <= value <= max_val):
+                    errors.append(
+                        f"[ERROR] {preset_name}.{param} = {value} "
+                        f"(valid: {min_val}-{max_val})"
+                    )
+    
+    if errors:
+        for error in errors:
+            print(error)
+        raise ValueError("Invalid preset configuration(s)")
+    
+    print("[OK] All presets validated")
+
+
+# Run validation on import
+validate_presets()
+
+print(f"[CONFIG] Loaded configuration with InstantID support")
 print(f"  Device: {device}")
 print(f"  Dtype: {dtype}")
 print(f"  Model Repo: {MODEL_REPO}")
-print(f"  HuggingFace Token: {'Set' if HUGGINGFACE_TOKEN else 'Not set (using IP-based access)'}")
+print(f"  InstantID Repo: {INSTANTID_CONFIG['repo']}")
+print(f"  HuggingFace Token: {'Set' if HUGGINGFACE_TOKEN else 'Not set (using IP-based access)'}")