Update generator.py

generator.py

@@ -34,6 +34,7 @@ class Generator:
         guidance_scale=1.5,
         num_inference_steps=6,
         img2img_strength=0.3,
+        face_strength=0.3,
         depth_strength=0.3,
         lineart_strength=0.3,
         seed=-1
@@ -82,26 +83,24 @@ class Generator:
 
             # Create keypoint image
             face_kps = draw_kps(processed_image, face_info['kps'])
-            face_guidance = (depth_strength + lineart_strength) / 2
-            
             # 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(face_guidance / 1.75)
+            controlnet_conditioning_scale = [face_strength, depth_strength, lineart_strength] 
+            self.mh.pipeline.set_ip_adapter_scale(0.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
+            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_guidance / 1.25, depth_strength/1.5, lineart_strength/1.5] 
+        control_guidance_end = [face_strength * 0.90, depth_strength * 0.70, lineart_strength * 0.80] 
 
         # --- Seed/Generator Logic ---
         if seed == -1 or seed is None:
@@ -129,7 +128,7 @@ class Generator:
             controlnet_conditioning_scale=controlnet_conditioning_scale,
             control_guidance_end=control_guidance_end,
             
-            clip_skip=1,
+            clip_skip=2,
             
         ).images[0]