Update generator.py

generator.py

@@ -756,13 +756,15 @@ class RetroArtConverter:
             try:
                 print("Encoding prompts with Compel...")
                 
-                # --- FIX 1: Move encoders to GPU ---
+                # --- FIX: Move text encoders to GPU for Compel ---
                 self.pipe.text_encoder.to(self.device)
                 self.pipe.text_encoder_2.to(self.device)
+                # --- END FIX ---
                 
-                # --- FIX 2: Tell Compel to use the GPU for its tensors ---
-                conditioning = self.compel(prompt, device=self.device)
-                negative_conditioning = self.compel(negative_prompt, device=self.device)
+                # --- FIX: Remove 'device=self.device' argument ---
+                conditioning = self.compel(prompt)
+                negative_conditioning = self.compel(negative_prompt)
+                # --- END FIX ---
                 
                 pipe_kwargs["prompt_embeds"] = conditioning[0]
                 pipe_kwargs["pooled_prompt_embeds"] = conditioning[1]
@@ -772,15 +774,18 @@ class RetroArtConverter:
                 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
             finally:
-                # --- FIX 3: Move encoders back to CPU ---
+                # --- FIX: Move text encoders back to CPU to save VRAM ---
                 try:
                     self.pipe.text_encoder.to("cpu")
                     self.pipe.text_encoder_2.to("cpu")
                 except Exception as e:
                     print(f"Could not move text encoders back to CPU: {e}")
+                # --- END FIX ---
         else:
             pipe_kwargs["prompt"] = prompt
             pipe_kwargs["negative_prompt"] = negative_prompt