Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ check_min_version("0.30.2")
|
|
| 14 |
|
| 15 |
|
| 16 |
# Build pipeline
|
| 17 |
-
controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-
|
| 18 |
transformer = FluxTransformer2DModel.from_pretrained(
|
| 19 |
"black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dytpe=torch.bfloat16
|
| 20 |
)
|
|
@@ -29,7 +29,7 @@ pipe.controlnet.to(torch.bfloat16)
|
|
| 29 |
|
| 30 |
|
| 31 |
MARKDOWN = """
|
| 32 |
-
# FLUX.1-dev-Inpainting-Model-
|
| 33 |
"""
|
| 34 |
|
| 35 |
@spaces.GPU()
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
# Build pipeline
|
| 17 |
+
controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)
|
| 18 |
transformer = FluxTransformer2DModel.from_pretrained(
|
| 19 |
"black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dytpe=torch.bfloat16
|
| 20 |
)
|
|
|
|
| 29 |
|
| 30 |
|
| 31 |
MARKDOWN = """
|
| 32 |
+
# FLUX.1-dev-Inpainting-Model-BETA-GPU 🔥
|
| 33 |
"""
|
| 34 |
|
| 35 |
@spaces.GPU()
|