Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
85f2c6e
1
Parent(s):
29dc892
Update app.py
Browse files
app.py
CHANGED
@@ -19,6 +19,10 @@ pipe = FluxControlNetInpaintingPipeline.from_pretrained(
|
|
19 |
transformer=transformer,
|
20 |
torch_dtype=torch.bfloat16
|
21 |
).to("cuda")
|
|
|
|
|
|
|
|
|
22 |
pipe.transformer.to(torch.bfloat16)
|
23 |
pipe.controlnet.to(torch.bfloat16)
|
24 |
|
|
|
19 |
transformer=transformer,
|
20 |
torch_dtype=torch.bfloat16
|
21 |
).to("cuda")
|
22 |
+
repo_name = "ByteDance/Hyper-SD"
|
23 |
+
ckpt_name = "Hyper-FLUX.1-dev-8steps-lora.safetensors"
|
24 |
+
pipe.load_lora_weights(hf_hub_download(repo_name, ckpt_name))
|
25 |
+
pipe.fuse_lora(lora_scale=0.125)
|
26 |
pipe.transformer.to(torch.bfloat16)
|
27 |
pipe.controlnet.to(torch.bfloat16)
|
28 |
|