Commit
•
585249a
1
Parent(s):
4e7780d
Update app.py
Browse files
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-
|
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="
|
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:
|