OmPrakashSingh1704 commited on
Commit
64496a7
·
1 Parent(s): f66121e
options/Banner_Model/Image2Image.py CHANGED
@@ -2,9 +2,11 @@ import imageio
2
  import numpy as np
3
  from PIL import Image
4
  import torch
5
- from .controlnet_flux import FluxControlNetModel
6
- from .transformer_flux import FluxTransformer2DModel
7
- from .pipeline_flux_controlnet_inpaint import FluxControlNetInpaintingPipeline
 
 
8
 
9
 
10
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
2
  import numpy as np
3
  from PIL import Image
4
  import torch
5
+ # from .controlnet_flux import FluxControlNetModel
6
+ # from .transformer_flux import FluxTransformer2DModel
7
+ # from .pipeline_flux_controlnet_inpaint import FluxControlNetInpaintingPipeline
8
+ from typing import Tuple
9
+ from diffusers import FluxInpaintPipeline
10
 
11
 
12
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
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