Hugging Face's logo

Spaces:
Himanshu806
/
fluxInpaint
Running on Zero

App Files Community
Himanshu806 commited on
Commit
e297481
·
verified ·
1 Parent(s): 098289a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ MAX_IMAGE_SIZE = 2048
11
 
12
  # Load pipeline with VAE enabled
13
  pipe = FluxInpaintPipeline.from_pretrained(
14
- "black-forest-labs/FLUX.1-Fill-dev",
15
- torch_dtype=torch.bfloat16
16
  ).to("cuda")
17
  pipe.load_lora_weights("alvdansen/flux-koda")
18
  pipe.enable_lora()
 
11
 
12
  # Load pipeline with VAE enabled
13
  pipe = FluxInpaintPipeline.from_pretrained(
14
+ "black-forest-labs/FLUX.1-Fill-dev"
15
+ # torch_dtype=torch.bfloat16
16
  ).to("cuda")
17
  pipe.load_lora_weights("alvdansen/flux-koda")
18
  pipe.enable_lora()