Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -31,9 +31,9 @@ pipe = FluxWithCFGPipeline.from_pretrained(
|
|
31 |
"ostris/OpenFLUX.1", torch_dtype=dtype
|
32 |
)
|
33 |
# pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
pipe.to("cuda")
|
38 |
torch.cuda.empty_cache()
|
39 |
|
|
|
31 |
"ostris/OpenFLUX.1", torch_dtype=dtype
|
32 |
)
|
33 |
# pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
|
34 |
+
pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
|
35 |
+
pipe.set_adapters("fast")
|
36 |
+
pipe.fuse_lora(adapter_names=["fast"], lora_scale=1.0)
|
37 |
pipe.to("cuda")
|
38 |
torch.cuda.empty_cache()
|
39 |
|