Update generator.py

generator.py

@@ -66,6 +66,7 @@ class Generator:
         
         # 5. Logic for Face vs No-Face (NOW INCLUDES KPS)
         # ControlNet order: [InstantID_KPS, Zoe_Depth, LineArt]
+        
         if face_info is not None:
             print("Face detected: Applying InstantID with keypoints.")
             
@@ -82,17 +83,17 @@ class Generator:
             # Create keypoint image
             face_kps = draw_kps(processed_image, face_info['kps'])
             face_guidance = (depth_strength + lineart_strength) / 2
-            face_kps_guidance_end = face_guidance / 1.5
             
             # Set strengths (using 0.8 from file's example)
             controlnet_conditioning_scale = [face_guidance, depth_strength, lineart_strength] 
-            self.mh.pipeline.set_ip_adapter_scale(0.0)
+            self.mh.pipeline.set_ip_adapter_scale(face_guidance / 1.75)
         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_guidance = 0.001
             face_kps_guidance_end = 0.001
             
             # Set strengths
@@ -100,7 +101,7 @@ class Generator:
             self.mh.pipeline.set_ip_adapter_scale(0.0)
 
         # We keep the guidance_end for pose low
-        control_guidance_end = [face_kps_guidance_end, depth_strength/1.5, lineart_strength/1.5] 
+        control_guidance_end = [face_guidance / 1.25, depth_strength/1.5, lineart_strength/1.5] 
 
         # --- Seed/Generator Logic ---
         if seed == -1 or seed is None: