Spaces:

Himanshu806
/

fluxInpaint

Running on Zero

ameerazam08 commited on Oct 10, 2024

Commit

1b2bb34

•

1 Parent(s): 099273d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ check_min_version("0.30.2")
 # Build pipeline
-controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Alpha", torch_dtype=torch.bfloat16)
 transformer = FluxTransformer2DModel.from_pretrained(
         "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dytpe=torch.bfloat16
     )
@@ -29,7 +29,7 @@ pipe.controlnet.to(torch.bfloat16)
 MARKDOWN = """
-# FLUX.1-dev-Inpainting-Model-Alpha-GPU 🔥
 """
 @spaces.GPU()

 # Build pipeline
+controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)
 transformer = FluxTransformer2DModel.from_pretrained(
         "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dytpe=torch.bfloat16
     )
 MARKDOWN = """
+# FLUX.1-dev-Inpainting-Model-BETA-GPU 🔥
 """
 @spaces.GPU()