Update generator.py

generator.py

@@ -305,9 +305,9 @@ class RetroArtConverter:
             # Generate depth map
             depth_image = self.get_depth_map(resized_image)
             
-            # ═══════════════════════════════════════════════════════════
+            # ═══════════════════════════════════════════════════════════
             # FACE DETECTION
-            # ═══════════════════════════════════════════════════════════
+            # ═══════════════════════════════════════════════════════════
             has_detected_faces = False
             face_kps_image = None
             face_embeddings = None
@@ -355,15 +355,14 @@ class RetroArtConverter:
                 except Exception as e:
                     print(f"[LORA] Could not fuse: {e}")
             
-            # ═══════════════════════════════════════════════════════════
+            # ═══════════════════════════════════════════════════════════
             # PIPELINE CONFIGURATION
-            # ═══════════════════════════════════════════════════════════
+            # ════════════��══════════════════════════════════════════════
             pipe_kwargs = {
                 "image": resized_image,
                 "strength": strength,
                 "num_inference_steps": num_inference_steps,
                 "guidance_scale": guidance_scale,
-                "cross_attention_kwargs": {"scale": lora_scale},
             }
             
             # Setup generator with seed
@@ -398,14 +397,14 @@ class RetroArtConverter:
                 pipe_kwargs["prompt"] = prompt
                 pipe_kwargs["negative_prompt"] = negative_prompt
             
-            # ═══════════════════════════════════════════════════════════
+            # ═══════════════════════════════════════════════════════════
             # CONTROLNET + IP-ADAPTER CONFIGURATION
-            # ═══════════════════════════════════════════════════════════
+            # ═══════════════════════════════════════════════════════════
             
             if has_detected_faces and face_kps_image is not None and face_embeddings is not None:
-                print("═" * 60)
+                print("═" * 60)
                 print("MODE: InstantID (Face Keypoints + Depth + IP-Adapter)")
-                print("═" * 60)
+                print("═" * 60)
                 
                 # Set IP-Adapter scale
                 self.pipe.set_ip_adapter_scale(identity_preservation)
@@ -430,13 +429,13 @@ class RetroArtConverter:
                 print(f"  [CONTROLNET] Identity scale: {identity_control_scale}")
                 print(f"  [CONTROLNET] Depth scale: {depth_control_scale}")
                 print(f"  [EMBEDDINGS] Shape: {face_embeddings.shape} (raw)")
-                print("  [INFO] Pipeline will handle: Resampler → Concatenation → Attention")
-                print("═" * 60)
+                print("  [INFO] Pipeline will handle: Resampler → Concatenation → Attention")
+                print("═" * 60)
             
             elif has_detected_faces and face_kps_image is not None:
-                print("═" * 60)
+                print("═" * 60)
                 print("MODE: InstantID Keypoints Only (no embeddings)")
-                print("═" * 60)
+                print("═" * 60)
                 
                 # Disable IP-Adapter
                 self.pipe.set_ip_adapter_scale(0.0)
@@ -456,12 +455,12 @@ class RetroArtConverter:
                 pipe_kwargs["image_embeds"] = zero_embeddings
                 
                 print("  [INFO] Using keypoints for structure only (zero embeddings)")
-                print("═" * 60)
+                print("═" * 60)
             
             else:
-                print("═" * 60)
+                print("═" * 60)
                 print("MODE: Depth Only (no face detection)")
-                print("═" * 60)
+                print("═" * 60)
                 
                 # Disable IP-Adapter
                 self.pipe.set_ip_adapter_scale(0.0)
@@ -479,20 +478,20 @@ class RetroArtConverter:
                 
                 print(f"  [CONTROLNET] Depth scale: {depth_control_scale}")
                 print("  [INFO] Generating without face preservation (zero embeddings)")
-                print("═" * 60)
+                print("═" * 60)
             
-            # ═══════════════════════════════════════════════════════════
+            # ═══════════════════════════════════════════════════════════
             # GENERATION
-            # ═══════════════════════════════════════════════════════════
+            # ═══════════════════════════════════════════════════════════
             print(f"\nGenerating: Steps={num_inference_steps}, CFG={guidance_scale}, Strength={strength}")
             
             result = self.pipe(**pipe_kwargs)
             
             generated_image = result.images[0]
             
-            # ═══════════════════════════════════════════════════════════
+            # ════════════════════════��══════════════════════════════════
             # POST-PROCESSING
-            # ═══════════════════════════════════════════════════════════
+            # ═══════════════════════════════════════════════════════════
             if enable_color_matching and has_detected_faces:
                 print("Applying enhanced face-aware color matching...")
                 try: