vilarin commited on
Commit
05fa434
·
verified ·
1 Parent(s): 4b42ada

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -11,6 +11,8 @@ model = "black-forest-labs/FLUX.1-dev"
11
  if torch.cuda.is_available():
12
  transformer = FluxTransformer2DModel.from_single_file(
13
  "https://huggingface.co/black-forest-labs/FLUX.1-Fill-dev/blob/main/flux1-fill-dev.safetensors",
 
 
14
  torch_dtype=torch.bfloat16
15
  )
16
  pipe = FluxInpaintPipeline.from_pretrained(
 
11
  if torch.cuda.is_available():
12
  transformer = FluxTransformer2DModel.from_single_file(
13
  "https://huggingface.co/black-forest-labs/FLUX.1-Fill-dev/blob/main/flux1-fill-dev.safetensors",
14
+ low_cpu_mem_usage=False,
15
+ ignore_mismatched_sizes=True,
16
  torch_dtype=torch.bfloat16
17
  )
18
  pipe = FluxInpaintPipeline.from_pretrained(