Update generator.py

generator.py

@@ -763,19 +763,19 @@ class RetroArtConverter:
         
         pipe_kwargs["generator"] = generator
         
-        # --- FIX: Reverted Compel block ---
-        # No more try/finally, no more .to(device)
-        # This works because optimize_pipeline() no longer offloads the text encoders.
+        # --- START FIX: Call Compel in a single batch ---
         if self.use_compel and self.compel is not None:
             try:
                 print("Encoding prompts with Compel...")
-                conditioning = self.compel(prompt)
-                negative_conditioning = self.compel(negative_prompt)
                 
-                pipe_kwargs["prompt_embeds"] = conditioning[0]
-                pipe_kwargs["pooled_prompt_embeds"] = conditioning[1]
-                pipe_kwargs["negative_prompt_embeds"] = negative_conditioning[0]
-                pipe_kwargs["negative_pooled_prompt_embeds"] = negative_conditioning[1]
+                # 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]
                 
                 print("[OK] Using Compel-encoded prompts")
             except Exception as e: