Spaces:
Running
on
A10G
Running
on
A10G
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,
|