Hugging Face's logo

Spaces:
Akjava
/
flux1-schnell-mask-inpaint
Running on Zero

App Files Community
Akjava commited on
Commit
97398f7
·
1 Parent(s): 451ef20

fix device

Browse files
Files changed (1) hide show
  1. flux1_inpaint.py +1 -1
flux1_inpaint.py CHANGED
@@ -43,7 +43,7 @@ def process_image(image,mask_image,prompt="a girl",negative_prompt="",model_id="
43
  return None
44
 
45
  pipe = FluxInpaintPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
46
-
47
  #batch_size =1
48
  generators = []
49
  generator = torch.Generator(device).manual_seed(seed)
 
43
  return None
44
 
45
  pipe = FluxInpaintPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
46
+ pipe.to(device)
47
  #batch_size =1
48
  generators = []
49
  generator = torch.Generator(device).manual_seed(seed)