Spaces:
Runtime error
Runtime error
File size: 1,709 Bytes
5fbd25d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
# -*- coding: utf-8 -*-
"""
Download models from url
@file: model_loader.py
@author: Konie
@update: 2024-03-22
"""
from modules.model_loader import load_file_from_url
def download_models():
"""
Download models from config
"""
vae_approx_filenames = [
('xlvaeapp.pth', 'https://huggingface.co/lllyasviel/misc/resolve/main/xlvaeapp.pth'),
('vaeapp_sd15.pth', 'https://huggingface.co/lllyasviel/misc/resolve/main/vaeapp_sd15.pt'),
('xl-to-v1_interposer-v3.1.safetensors', 'https://huggingface.co/lllyasviel/misc/resolve/main/xl-to-v1_interposer-v3.1.safetensors')
]
from modules.config import (
paths_checkpoints as modelfile_path,
paths_loras as lorafile_path,
path_vae_approx as vae_approx_path,
path_fooocus_expansion as fooocus_expansion_path,
path_embeddings as embeddings_path,
checkpoint_downloads,
embeddings_downloads,
lora_downloads)
for file_name, url in checkpoint_downloads.items():
load_file_from_url(url=url, model_dir=modelfile_path[0], file_name=file_name)
for file_name, url in embeddings_downloads.items():
load_file_from_url(url=url, model_dir=embeddings_path, file_name=file_name)
for file_name, url in lora_downloads.items():
load_file_from_url(url=url, model_dir=lorafile_path[0], file_name=file_name)
for file_name, url in vae_approx_filenames:
load_file_from_url(url=url, model_dir=vae_approx_path, file_name=file_name)
load_file_from_url(
url='https://huggingface.co/lllyasviel/misc/resolve/main/fooocus_expansion.bin',
model_dir=fooocus_expansion_path,
file_name='pytorch_model.bin'
)
|