mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-03-12 08:30:27 +00:00
option: reduce connecting to huggingface
for assets if local cache is available note: enabling this with prevent the assets from being updated
This commit is contained in:
41
modules/patch_hf_hub_download.py
Normal file
41
modules/patch_hf_hub_download.py
Normal file
@@ -0,0 +1,41 @@
|
||||
from modules.patches import patch
|
||||
from modules.errors import report
|
||||
from inspect import signature
|
||||
from functools import wraps
|
||||
|
||||
try:
|
||||
from huggingface_hub.utils import LocalEntryNotFoundError
|
||||
from huggingface_hub import file_download
|
||||
|
||||
def try_local_files_only(func):
|
||||
if (param := signature(func).parameters.get('local_files_only', None)) and not param.kind == param.KEYWORD_ONLY:
|
||||
raise ValueError(f'{func.__name__} does not have keyword-only parameter "local_files_only"')
|
||||
|
||||
@wraps(func)
|
||||
def wrapper(*args, **kwargs):
|
||||
try:
|
||||
from modules.shared import opts
|
||||
try_offline_mode = not kwargs.get('local_files_only') and opts.hd_dl_local_first
|
||||
except Exception:
|
||||
report('Error in try_local_files_only - skip try_local_files_only', exc_info=True)
|
||||
try_offline_mode = False
|
||||
|
||||
if try_offline_mode:
|
||||
try:
|
||||
return func(*args, **{**kwargs, 'local_files_only': True})
|
||||
except LocalEntryNotFoundError:
|
||||
pass
|
||||
except Exception:
|
||||
report('Unexpected exception in try_local_files_only - retry without patch', exc_info=True)
|
||||
|
||||
return func(*args, **kwargs)
|
||||
|
||||
return wrapper
|
||||
|
||||
try:
|
||||
patch(__name__, file_download, 'hf_hub_download', try_local_files_only(file_download.hf_hub_download))
|
||||
except RuntimeError:
|
||||
pass # already patched
|
||||
|
||||
except Exception:
|
||||
report('Error patching hf_hub_download', exc_info=True)
|
||||
Reference in New Issue
Block a user