Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -24,6 +24,10 @@ pipe = FluxWithCFGPipeline.from_pretrained(
|
|
24 |
)
|
25 |
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
|
26 |
pipe.to("cuda")
|
|
|
|
|
|
|
|
|
27 |
|
28 |
torch.cuda.empty_cache()
|
29 |
|
|
|
24 |
)
|
25 |
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
|
26 |
pipe.to("cuda")
|
27 |
+
pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v2.3.safetensors', adapter_name="better")
|
28 |
+
pipe.set_adapters(["better"], adapter_weights=[1.0])
|
29 |
+
pipe.fuse_lora(adapter_name=["better"], lora_scale=1.0)
|
30 |
+
pipe.unload_lora_weights()
|
31 |
|
32 |
torch.cuda.empty_cache()
|
33 |
|