AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

upload & fix

Browse Source
This commit is contained in:
Patrick von Platen
2022-06-09 10:58:50 +00:00
parent 2fa1d64841
commit 7ba3130cc2
1 changed files with 17 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/diffusers/dynamic_modules_utils.py
View File

@@ -209,21 +209,24 @@ def get_cached_module_file(
module_file_or_url = os.path.join(pretrained_model_name_or_path, module_file)
submodule = "local"
try:
# Load from URL or cache if already cached
resolved_module_file = cached_download(
module_file_or_url,
cache_dir=cache_dir,
force_download=force_download,
proxies=proxies,
resume_download=resume_download,
local_files_only=local_files_only,
use_auth_token=use_auth_token,
)
if os.path.isfile(module_file_or_url):
resolved_module_file = module_file_or_url
else:
try:
# Load from URL or cache if already cached
resolved_module_file = cached_download(
module_file_or_url,
cache_dir=cache_dir,
force_download=force_download,
proxies=proxies,
resume_download=resume_download,
local_files_only=local_files_only,
use_auth_token=use_auth_token,
)
except EnvironmentError:
logger.error(f"Could not locate the {module_file} inside {pretrained_model_name_or_path}.")
raise
except EnvironmentError:
logger.error(f"Could not locate the {module_file} inside {pretrained_model_name_or_path}.")
raise
# Check we have all the requirements in our environment
modules_needed = check_imports(resolved_module_file)