diff --git a/launch.py b/launch.py index fc9bf6c..3da5b8d 100644 --- a/launch.py +++ b/launch.py @@ -5,7 +5,7 @@ import platform from modules.launch_util import commit_hash, fooocus_tag, is_installed, run, python, \ run_pip, repo_dir, git_clone, requirements_met, script_path, dir_repos from modules.model_loader import load_file_from_url -from modules.path import modelfile_path +from modules.path import modelfile_path, lorafile_path REINSTALL_ALL = False @@ -61,10 +61,17 @@ model_filenames = [ 'https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/resolve/main/sd_xl_refiner_1.0.safetensors') ] +lora_filenames = [ + ('sd_xl_offset_example-lora_1.0.safetensors', + 'https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_offset_example-lora_1.0.safetensors') +] + def download_models(): for file_name, url in model_filenames: load_file_from_url(url=url, model_dir=modelfile_path, file_name=file_name) + for file_name, url in lora_filenames: + load_file_from_url(url=url, model_dir=lorafile_path, file_name=file_name) return diff --git a/modules/path.py b/modules/path.py index 6c38f00..2d38161 100644 --- a/modules/path.py +++ b/modules/path.py @@ -1,3 +1,4 @@ import os modelfile_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../models/checkpoints/')) +lorafile_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../models/loras/'))