AndranikSargsyan commited on
Commit
b35c416
β€’
1 Parent(s): 44fecdc

change sd1.5 download paths

Browse files
Files changed (1) hide show
  1. 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': False,
10
- 'model_path': 'sd-1-5-inpainting/sd-v1-5-inpainting.ckpt',
11
- 'download_url': 'https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt?download=true'
 
 
 
 
 
 
 
 
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,