Spaces:
Paused
Paused
AndranikSargsyan
commited on
Commit
·
b35c416
1
Parent(s):
44fecdc
change sd1.5 download paths
Browse files- src/models/inpainting.py +11 -3
src/models/inpainting.py
CHANGED
|
@@ -6,9 +6,17 @@ from .common import MODEL_FOLDER, load_sd_inpainting_model, download_file
|
|
| 6 |
model_dict = {
|
| 7 |
'sd15_inp': {
|
| 8 |
'sd_version': 1,
|
| 9 |
-
'diffusers_ckpt':
|
| 10 |
-
'model_path':
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
},
|
| 13 |
'ds8_inp': {
|
| 14 |
'sd_version': 1,
|
|
|
|
| 6 |
model_dict = {
|
| 7 |
'sd15_inp': {
|
| 8 |
'sd_version': 1,
|
| 9 |
+
'diffusers_ckpt': True,
|
| 10 |
+
'model_path': OrderedDict([
|
| 11 |
+
('unet', 'sd-1-5-inpainting/unet.fp16.safetensors'),
|
| 12 |
+
('encoder', 'sd-1-5-inpainting/encoder.fp16.safetensors'),
|
| 13 |
+
('vae', 'sd-1-5-inpainting/vae.fp16.safetensors')
|
| 14 |
+
]),
|
| 15 |
+
'download_url': OrderedDict([
|
| 16 |
+
('unet', 'https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/unet/diffusion_pytorch_model.fp16.safetensors?download=true'),
|
| 17 |
+
('encoder', 'https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/text_encoder/model.fp16.safetensors?download=true'),
|
| 18 |
+
('vae', 'https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/vae/diffusion_pytorch_model.fp16.safetensors?download=true')
|
| 19 |
+
])
|
| 20 |
},
|
| 21 |
'ds8_inp': {
|
| 22 |
'sd_version': 1,
|