Update generator.py

generator.py

@@ -840,7 +840,20 @@ class RetroArtConverter:
                             
                             combined_embeds = torch.cat([original_embeds, face_proj_embeds], dim=1)
                             pipe_kwargs['prompt_embeds'] = combined_embeds
-                            print(f"  [OK] Face embeddings concatenated successfully! New shape: {combined_embeds.shape}")
+                            
+                            # FIX: Also pad negative_prompt_embeds to match the new shape
+                            if 'negative_prompt_embeds' in pipe_kwargs:
+                                negative_embeds = pipe_kwargs['negative_prompt_embeds']
+                                # Create zero padding with same shape as face_proj_embeds
+                                padding = torch.zeros(
+                                    (negative_embeds.shape[0], face_proj_embeds.shape[1], negative_embeds.shape[2]),
+                                    device=negative_embeds.device,
+                                    dtype=negative_embeds.dtype
+                                )
+                                pipe_kwargs['negative_prompt_embeds'] = torch.cat([negative_embeds, padding], dim=1)
+                                print(f"  [OK] Negative prompt padded to match: {pipe_kwargs['negative_prompt_embeds'].shape}")
+                            
+                            print(f"  [OK] Face embeddings concatenated successfully! Prompt: {combined_embeds.shape}, Negative: {pipe_kwargs['negative_prompt_embeds'].shape}")
                         else:
                             print(f"  [WARNING] Can't concatenate - no prompt_embeds (use Compel)")