ameerazam08 commited on
Commit
1b2bb34
1 Parent(s): 099273d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-Alpha", torch_dtype=torch.bfloat16)
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-Alpha-GPU 🔥
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()