KingNish commited on
Commit
9a1981a
·
verified ·
1 Parent(s): 35f3b0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ pipe = FluxWithCFGPipeline.from_pretrained(
22
  pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
23
  pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
24
  pipe.set_adapters("fast")
25
- pipeline.fuse_lora(adapter_names=["fast"], lora_scale=1.0)
26
  pipe.to("cuda")
27
  # pipe.transformer.to(memory_format=torch.channels_last)
28
  # pipe.transformer = torch.compile(
 
22
  pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
23
  pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
24
  pipe.set_adapters("fast")
25
+ pipe.fuse_lora(adapter_names=["fast"], lora_scale=1.0)
26
  pipe.to("cuda")
27
  # pipe.transformer.to(memory_format=torch.channels_last)
28
  # pipe.transformer = torch.compile(