Update generator.py

generator.py

@@ -71,7 +71,6 @@ class Generator:
         if face_info is not None:
             print("Face detected: Applying InstantID with keypoints.")
             
-            # --- FIX APPLIED HERE ---
             # We use face_info['embedding'] (raw) instead of normed_embedding.
             # Raw embedding has higher magnitude (~20-30) required for the adapter.
             face_emb = torch.tensor(
@@ -79,28 +78,41 @@ class Generator:
                 dtype=Config.DTYPE,
                 device=Config.DEVICE
             ).unsqueeze(0)
-            # --- END FIX ---
 
             # Create keypoint image
             face_kps = draw_kps(processed_image, face_info['kps'])
-            # Set strengths (using 0.8 from file's example)
+            
+            # Set strengths
             controlnet_conditioning_scale = [face_strength, depth_strength, lineart_strength] 
-            self.mh.pipeline.set_ip_adapter_scale(0.8)
+            
+            # --- UPDATED: Reduced IP Adapter Scale ---
+            # Lowered from 0.8 to 0.7 to allow LoRA style (pixel art) to 
+            # override realistic skin textures while keeping identity.
+            self.mh.pipeline.set_ip_adapter_scale(0.7)
         else:
             print("No face detected: Disabling InstantID.")
             # Create dummy embedding
             face_emb = torch.zeros((1, 512), dtype=Config.DTYPE, device=Config.DEVICE)
             # Create dummy keypoint image (black)
             face_kps = Image.new('RGB', (target_width, target_height), (0, 0, 0))
-            face_kps_guidance_end = 0.001
-            face_strength_end = 0.001
             
             # Set strengths
             controlnet_conditioning_scale = [0.0, depth_strength, lineart_strength] 
             self.mh.pipeline.set_ip_adapter_scale(0.0)
 
-        # We keep the guidance_end for pose low
-        control_guidance_end = [face_strength * 0.4, depth_strength * 0.8, lineart_strength * 0.6] 
+        # --- UPDATED: Control Guidance End Strategy ---
+        # We cap the Face ControlNet duration. 
+        # Even if strength is 1.0, we stop it at 0.6 (60%) of the steps.
+        # This leaves the final 40% of steps pure for the Pixel Art LoRA 
+        # to "pixelize" the face without the ControlNet trying to fix it back to a photo.
+        
+        face_end_step = min(0.6, face_strength)
+        
+        control_guidance_end = [
+            face_end_step,      # InstantID: Stop early for style
+            depth_strength,     # Depth: Keep structure longer
+            lineart_strength    # Lineart: Keep outlines longer
+        ] 
 
         # --- Seed/Generator Logic ---
         if seed == -1 or seed is None: