multimodalart HF staff commited on
Commit
46d23be
1 Parent(s): 43d2e9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,8 +25,8 @@ pipe = FluxControlNetInpaintingPipeline.from_pretrained(
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,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">FLUX.1-dev-Controlnet-Inpainting-Beta</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:
 
25
  torch_dtype=torch.bfloat16
26
  )
27
 
28
+ repo_name = "ByteDance/Hyper-SD"
29
+ ckpt_name = "Hyper-FLUX.1-dev-8steps-lora.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/ByteDance/Hyper-SD/blob/main/Hyper-FLUX.1-dev-8steps-lora.safetensors" target="_blank">Hyper-FLUX.1-dev-8steps-lora</a></div>
225
  """
226
 
227
  with gr.Blocks(css=css) as demo: