LibreFLUX_LoRAs_Gallery

Running on Zero

AlekseyCalvin commited on Oct 17

Commit

daabd5f

•

1 Parent(s): 01424e7

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -561,7 +561,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
         else:
             batch_size = prompt_embeds.shape[0]
-        cfg.device = get_device()
         lora_scale = (
             self.joint_attention_kwargs.get("scale", None) if self.joint_attention_kwargs is not None else None

         else:
             batch_size = prompt_embeds.shape[0]
+        device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
         lora_scale = (
             self.joint_attention_kwargs.get("scale", None) if self.joint_attention_kwargs is not None else None