Update generator.py

generator.py

@@ -69,16 +69,19 @@ class Generator:
         if face_info is not None:
             print("Face detected: Applying InstantID with keypoints.")
             
-            # --- MODIFIED: Explicitly cast tensor to Config.DTYPE ---
+            # --- 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(
-                face_info.normed_embedding, 
-                dtype=Config.DTYPE,  # <-- This is the fix
+                face_info['embedding'], 
+                dtype=Config.DTYPE,
                 device=Config.DEVICE
             ).unsqueeze(0)
-            # --- END MODIFIED ---
+            # --- END FIX ---
 
             # Create keypoint image
             face_kps = draw_kps(processed_image, face_info['kps'])
+            face_kps_guidance_end = 0.3
             
             # Set strengths (using 0.8 from file's example)
             controlnet_conditioning_scale = [0.8, depth_strength, lineart_strength] 
@@ -89,13 +92,14 @@ class Generator:
             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.0
             
             # 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 = [0.3, 0.6, 0.6] 
+        control_guidance_end = [face_kps_guidance_end, 0.6, 0.6] 
 
         # --- Seed/Generator Logic ---
         if seed == -1 or seed is None:
@@ -123,7 +127,7 @@ class Generator:
             controlnet_conditioning_scale=controlnet_conditioning_scale,
             control_guidance_end=control_guidance_end,
             
-            clip_skip=2,
+            clip_skip=1,
             
         ).images[0]