Update generator.py

generator.py

@@ -8,6 +8,7 @@ import cv2
 from PIL import Image
 import torch.nn.functional as F
 from torchvision import transforms
+import traceback
 
 from config import (
     device, dtype, TRIGGER_WORD, MULTI_SCALE_FACTORS,
@@ -676,7 +677,6 @@ class RetroArtConverter:
                         
                 except Exception as e:
                     print(f"[ERROR] InsightFace detection failed: {e}")
-                    import traceback
                     traceback.print_exc()
             else:
                 print("[INFO] InsightFace not available (face_app is None)")
@@ -705,7 +705,6 @@ class RetroArtConverter:
                             print("✗ MediapipeFace found no faces")
                     except Exception as e:
                         print(f"[ERROR] MediapipeFace detection failed: {e}")
-                        import traceback
                         traceback.print_exc()
                 else:
                     print("[INFO] MediapipeFaceDetector not available")
@@ -781,7 +780,6 @@ class RetroArtConverter:
                 print("[OK] Using Compel-encoded prompts")
             except Exception as e:
                 print(f"Compel encoding failed, using standard prompts: {e}")
-                import traceback
                 traceback.print_exc()
                 pipe_kwargs["prompt"] = prompt
                 pipe_kwargs["negative_prompt"] = negative_prompt
@@ -840,20 +838,7 @@ class RetroArtConverter:
                             
                             combined_embeds = torch.cat([original_embeds, face_proj_embeds], dim=1)
                             pipe_kwargs['prompt_embeds'] = combined_embeds
-                            
-                            # 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}")
+                            print(f"  [OK] Face embeddings concatenated successfully! New shape: {combined_embeds.shape}")
                         else:
                             print(f"  [WARNING] Can't concatenate - no prompt_embeds (use Compel)")