multimodalart HF staff commited on
Commit
585249a
1 Parent(s): 4e7780d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ from huggingface_hub import hf_hub_download
13
  from optimum.quanto import freeze, qfloat8, quantize
14
 
15
 
16
- controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Alpha", torch_dtype=torch.bfloat16)
17
  transformer = FluxTransformer2DModel.from_pretrained(
18
  "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16
19
  )
@@ -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="https://huggingface.co/alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Alpha" target="_blank"><code>FLUX.1-dev-Controlnet-Inpainting-Beta</code></a></div>
225
  """
226
 
227
  with gr.Blocks(css=css) as demo:
 
13
  from optimum.quanto import freeze, qfloat8, quantize
14
 
15
 
16
+ controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)
17
  transformer = FluxTransformer2DModel.from_pretrained(
18
  "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16
19
  )
 
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: