Update generator.py

generator.py

@@ -704,7 +704,7 @@ class RetroArtConverter:
                         else:
                             print("✗ MediapipeFace found no faces")
                     except Exception as e:
-                        print(f"ERROR] MediapipeFace detection failed: {e}")
+                        print(f"[ERROR] MediapipeFace detection failed: {e}")
                         import traceback
                         traceback.print_exc()
                 else:
@@ -763,22 +763,21 @@ class RetroArtConverter:
         
         pipe_kwargs["generator"] = generator
         
-        # --- START FIX: Call Compel in a single batch ---
-        # Use Compel for prompt encoding if available
+        # --- START FIX: Correct Compel batching and slicing ---
         if self.use_compel and self.compel is not None:
             try:
                 print("Encoding prompts with Compel...")
-        
+                
                 # Pass both prompts as a list to be batched
                 conditioning_batch, pooled_batch = self.compel([prompt, negative_prompt])
-        
+
                 # Unpack the batch results using slicing
                 # [0:1] keeps the batch dimension, which is required
                 pipe_kwargs["prompt_embeds"] = conditioning_batch[0:1]
                 pipe_kwargs["pooled_prompt_embeds"] = pooled_batch[0:1]
                 pipe_kwargs["negative_prompt_embeds"] = conditioning_batch[1:2]
                 pipe_kwargs["negative_pooled_prompt_embeds"] = pooled_batch[1:2]
-        
+                
                 print("[OK] Using Compel-encoded prompts")
             except Exception as e:
                 print(f"Compel encoding failed, using standard prompts: {e}")