OmPrakashSingh1704 commited on
Commit
268b0bb
·
1 Parent(s): 4028c6c
options/Banner_Model/Image2Image.py CHANGED
@@ -29,7 +29,7 @@ def I2I(prompt, image, width=1024, height=1024, guidance_scale=8.0, num_inferenc
29
 
30
  pipe = FluxInpaintPipeline.from_pretrained(
31
  "black-forest-labs/FLUX.1-schnell",
32
- torch_dtype=torch.float16, variant="fp16").to(device)
33
 
34
  img_url, mask = dummy(image)
35
 
 
29
 
30
  pipe = FluxInpaintPipeline.from_pretrained(
31
  "black-forest-labs/FLUX.1-schnell",
32
+ torch_dtype=torch.float16).to(device)
33
 
34
  img_url, mask = dummy(image)
35
 
options/Banner_Model/__pycache__/Image2Image.cpython-310.pyc CHANGED
Binary files a/options/Banner_Model/__pycache__/Image2Image.cpython-310.pyc and b/options/Banner_Model/__pycache__/Image2Image.cpython-310.pyc differ