OmPrakashSingh1704 commited on
Commit
e0df40d
·
1 Parent(s): f0d07b8
options/Banner_Model/Image2Image.py CHANGED
@@ -1,7 +1,7 @@
1
  import imageio
2
  import numpy as np
3
  from PIL import Image
4
- from diffusers import AutoPipelineForInpainting
5
  import torch
6
 
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -27,8 +27,8 @@ def dummy(img):
27
 
28
  def I2I(prompt, image, width=1024, height=1024, guidance_scale=8.0, num_inference_steps=20, strength=0.99):
29
 
30
- pipe = AutoPipelineForInpainting.from_pretrained(
31
- "diffusers/stable-diffusion-xl-1.0-inpainting-0.1",
32
  torch_dtype=torch.float16, variant="fp16").to(device)
33
 
34
  img_url, mask = dummy(image)
 
1
  import imageio
2
  import numpy as np
3
  from PIL import Image
4
+ from diffusers import FluxInpaintPipeline
5
  import torch
6
 
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
27
 
28
  def I2I(prompt, image, width=1024, height=1024, guidance_scale=8.0, num_inference_steps=20, strength=0.99):
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)
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
 
options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc CHANGED
Binary files a/options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc and b/options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc differ
 
options/Banner_Model/__pycache__/Text2Banner.cpython-310.pyc CHANGED
Binary files a/options/Banner_Model/__pycache__/Text2Banner.cpython-310.pyc and b/options/Banner_Model/__pycache__/Text2Banner.cpython-310.pyc differ
 
options/Banner_Model/__pycache__/__init__.cpython-310.pyc CHANGED
Binary files a/options/Banner_Model/__pycache__/__init__.cpython-310.pyc and b/options/Banner_Model/__pycache__/__init__.cpython-310.pyc differ
 
options/Video_model/__pycache__/Model.cpython-310.pyc CHANGED
Binary files a/options/Video_model/__pycache__/Model.cpython-310.pyc and b/options/Video_model/__pycache__/Model.cpython-310.pyc differ
 
options/Video_model/__pycache__/__init__.cpython-310.pyc CHANGED
Binary files a/options/Video_model/__pycache__/__init__.cpython-310.pyc and b/options/Video_model/__pycache__/__init__.cpython-310.pyc differ
 
options/__pycache__/Banner.cpython-310.pyc CHANGED
Binary files a/options/__pycache__/Banner.cpython-310.pyc and b/options/__pycache__/Banner.cpython-310.pyc differ
 
options/__pycache__/Video.cpython-310.pyc CHANGED
Binary files a/options/__pycache__/Video.cpython-310.pyc and b/options/__pycache__/Video.cpython-310.pyc differ