Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
01424e7
1
Parent(s):
ca527e8
Update pipeline.py
Browse files- pipeline.py +1 -2
pipeline.py
CHANGED
|
@@ -561,8 +561,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
| 561 |
else:
|
| 562 |
batch_size = prompt_embeds.shape[0]
|
| 563 |
|
| 564 |
-
device =
|
| 565 |
-
self._execution_device = get_device()
|
| 566 |
|
| 567 |
lora_scale = (
|
| 568 |
self.joint_attention_kwargs.get("scale", None) if self.joint_attention_kwargs is not None else None
|
|
|
|
| 561 |
else:
|
| 562 |
batch_size = prompt_embeds.shape[0]
|
| 563 |
|
| 564 |
+
cfg.device = get_device()
|
|
|
|
| 565 |
|
| 566 |
lora_scale = (
|
| 567 |
self.joint_attention_kwargs.get("scale", None) if self.joint_attention_kwargs is not None else None
|