Spaces:
Runtime error
Runtime error
stable difussion inpainting v2
Browse files- requirements.txt +3 -3
- stablediffusion-infinity/app.py +1 -2
requirements.txt
CHANGED
@@ -1,8 +1,8 @@
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
2 |
torch
|
3 |
huggingface_hub
|
4 |
-
diffusers
|
5 |
-
transformers
|
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
|
|
|
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 |
-
"
|
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")
|