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:
|