Update generator.py

generator.py

@@ -764,20 +764,21 @@ class RetroArtConverter:
         pipe_kwargs["generator"] = generator
         
         # --- START FIX: Call Compel in a single batch ---
-       if self.use_compel and self.compel is not None:
+        # Use Compel for prompt encoding if available
+        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}")