Update custom_pipeline.py

custom_pipeline.py

@@ -88,7 +88,7 @@ class FluxWithCFGPipeline(FluxPipeline):
 
         # 2. Define call parameters
         batch_size = 1 if isinstance(prompt, str) else len(prompt)
-        device = self._execution_device
+        device = "cuda" if torch.cuda.is_available() else "cpu"
 
         # 3. Encode prompt
         lora_scale = joint_attention_kwargs.get("scale", None) if joint_attention_kwargs is not None else None