Update pipeline.py

pipeline.py

@@ -702,6 +702,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
         timesteps: List[int] = None,
         eta: Optional[float] = 0.0,
         guidance_scale: float = 3.5,
+        device: Optional[int] = None,
         num_images_per_prompt: Optional[int] = 1,
         generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
         latents: Optional[torch.FloatTensor] = None,
@@ -737,6 +738,8 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
         self._joint_attention_kwargs = joint_attention_kwargs
         self._interrupt = False
 
+        device = self._execution_device
+
         # 2. Define call parameters
         if prompt is not None and isinstance(prompt, str):
             batch_size = 1
@@ -745,8 +748,6 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
         else:
             batch_size = prompt_embeds.shape[0]
 
-        device = self._execution_device
-
         do_classifier_free_guidance = guidance_scale > 1.0
 
         lora_scale = (