* download upscaler only when user use it

* use better method
This commit is contained in:
lllyasviel
2023-10-08 00:28:03 -07:00
committed by GitHub
parent 6faaac333b
commit f170e1c08e
4 changed files with 22 additions and 12 deletions
-8
View File
@@ -71,12 +71,6 @@ vae_approx_filenames = [
]
upscaler_filenames = [
('fooocus_upscaler_s409985e5.bin',
'https://huggingface.co/lllyasviel/misc/resolve/main/fooocus_upscaler_s409985e5.bin')
]
def download_models():
for file_name, url in model_filenames:
load_file_from_url(url=url, model_dir=modelfile_path, file_name=file_name)
@@ -84,8 +78,6 @@ def download_models():
load_file_from_url(url=url, model_dir=lorafile_path, 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)
for file_name, url in upscaler_filenames:
load_file_from_url(url=url, model_dir=upscale_models_path, file_name=file_name)
load_file_from_url(
url='https://huggingface.co/lllyasviel/misc/resolve/main/fooocus_expansion.bin',