Update generator.py

generator.py

@@ -764,22 +764,25 @@ class RetroArtConverter:
         pipe_kwargs["generator"] = generator
         
         # --- START FIX: Call Compel in a single batch ---
-        if self.use_compel and self.compel is not None:
+       if self.use_compel and self.compel is not None:
             try:
                 print("Encoding prompts with Compel...")
                 
-                # Pass both prompts as a list
-                conditioning_batch = self.compel([prompt, negative_prompt])
-                
-                # Unpack the batch results
-                pipe_kwargs["prompt_embeds"] = conditioning_batch[0][0]
-                pipe_kwargs["pooled_prompt_embeds"] = conditioning_batch[0][1]
-                pipe_kwargs["negative_prompt_embeds"] = conditioning_batch[1][0]
-                pipe_kwargs["negative_pooled_prompt_embeds"] = conditioning_batch[1][1]
+                # 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}")
+                import traceback
+                traceback.print_exc()
                 pipe_kwargs["prompt"] = prompt
                 pipe_kwargs["negative_prompt"] = negative_prompt
         else: