Spaces:
Running
on
L40S
Running
on
L40S
Commit
•
fdeebfe
1
Parent(s):
f3d3190
Update app.py
Browse files
app.py
CHANGED
@@ -25,8 +25,8 @@ pipe = FluxControlNetInpaintingPipeline.from_pretrained(
|
|
25 |
torch_dtype=torch.bfloat16
|
26 |
)
|
27 |
|
28 |
-
repo_name = "
|
29 |
-
ckpt_name = "
|
30 |
pipe.load_lora_weights(hf_hub_download(repo_name, ckpt_name))
|
31 |
pipe.fuse_lora(lora_scale=0.125)
|
32 |
pipe.transformer.to(torch.bfloat16)
|
@@ -221,7 +221,7 @@ css = """
|
|
221 |
|
222 |
title = """<h1 align="center">FLUX Image Outpaint</h1>
|
223 |
<div align="center">Drop an image you would like to extend, pick your expected ratio and hit Generate.</div>
|
224 |
-
<div align="center">Using <a href="alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta" target="_blank"><code>FLUX.1-dev-Controlnet-Inpainting-Beta</code></a></div>
|
225 |
"""
|
226 |
|
227 |
with gr.Blocks(css=css) as demo:
|
|
|
25 |
torch_dtype=torch.bfloat16
|
26 |
)
|
27 |
|
28 |
+
repo_name = "alimama-creative/FLUX.1-Turbo-Alpha"
|
29 |
+
ckpt_name = "diffusion_pytorch_model.safetensors"
|
30 |
pipe.load_lora_weights(hf_hub_download(repo_name, ckpt_name))
|
31 |
pipe.fuse_lora(lora_scale=0.125)
|
32 |
pipe.transformer.to(torch.bfloat16)
|
|
|
221 |
|
222 |
title = """<h1 align="center">FLUX Image Outpaint</h1>
|
223 |
<div align="center">Drop an image you would like to extend, pick your expected ratio and hit Generate.</div>
|
224 |
+
<div align="center">Using <a href="alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta" target="_blank"><code>FLUX.1-dev-Controlnet-Inpainting-Beta</code></a> + <a href="https://huggingface.co/alimama-creative/FLUX.1-Turbo-Alpha" target="_blank">FLUX.1-Turbo-alpha 8-step LoRA</a></div>
|
225 |
"""
|
226 |
|
227 |
with gr.Blocks(css=css) as demo:
|