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")
|