radames commited on
Commit
3a90284
1 Parent(s): 0af39cb

stable difussion inpainting v2

Browse files
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch
3
  huggingface_hub
4
- diffusers==0.7.0
5
- transformers==4.23.1
6
  scikit-image==0.19.3
7
  Pillow==9.2.0
8
  opencv-python-headless==4.6.0.66
@@ -15,4 +15,4 @@ python-magic==0.4.27
15
  fastapi-utils==0.2.1
16
  shortuuid==1.0
17
  tqdm==4.64
18
- accelerate==0.13
 
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch
3
  huggingface_hub
4
+ git+https://github.com/huggingface/diffusers.git@9f476388
5
+ transformers
6
  scikit-image==0.19.3
7
  Pillow==9.2.0
8
  opencv-python-headless==4.6.0.66
 
15
  fastapi-utils==0.2.1
16
  shortuuid==1.0
17
  tqdm==4.64
18
+ accelerate
stablediffusion-infinity/app.py CHANGED
@@ -110,8 +110,7 @@ def get_model():
110
  if "inpaint" not in model:
111
  vae = AutoencoderKL.from_pretrained(f"stabilityai/sd-vae-ft-ema")
112
  inpaint = StableDiffusionInpaintPipeline.from_pretrained(
113
- "runwayml/stable-diffusion-inpainting",
114
- revision="fp16",
115
  torch_dtype=torch.float16,
116
  vae=vae,
117
  ).to("cuda")
 
110
  if "inpaint" not in model:
111
  vae = AutoencoderKL.from_pretrained(f"stabilityai/sd-vae-ft-ema")
112
  inpaint = StableDiffusionInpaintPipeline.from_pretrained(
113
+ "radames/stable-diffusion-v2-inpainting",
 
114
  torch_dtype=torch.float16,
115
  vae=vae,
116
  ).to("cuda")