Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

video loader use generic methods and auth

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-16 09:02:40 +02:00
parent 12bf006b00
commit cd25a5296a
7 changed files with 45 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+7 -5
View File
@@ -1605,27 +1605,29 @@ def unload_model_weights(op='model'):
log.debug(f'Unload {op}: {memory_stats()} fn={fn}')
def hf_auth_check(checkpoint_info: CheckpointInfo, force:bool=False):
def hf_auth_check(checkpoint_info: CheckpointInfo | str, force:bool=False):
if shared.opts.offline_mode:
log.info('Offline mode: skipping auth check')
return False
login = None
if not force:
try:
if (checkpoint_info.path.endswith('.safetensors') and os.path.isfile(checkpoint_info.path)): # skip check for single-file safetensors models
fn = checkpoint_info.path if isinstance(checkpoint_info, CheckpointInfo) else checkpoint_info
if (fn.endswith('.safetensors') and os.path.isfile(fn)): # skip check for single-file safetensors models
return True
if os.path.exists(checkpoint_info.path) and os.path.isdir(checkpoint_info.path) and any(os.path.isfile(os.path.join(checkpoint_info.path, f)) for f in ('model_index.json', 'modular_model_index.json')): # skip check for local diffusers folders
if os.path.exists(fn) and os.path.isdir(fn) and any(os.path.isfile(os.path.join(fn, f)) for f in ('model_index.json', 'modular_model_index.json')): # skip check for local diffusers folders
return True
except Exception:
pass
repo_id = path_to_repo(checkpoint_info)
repo_id = path_to_repo(checkpoint_info) # already handles str or CheckpointInfo
if repo_id is None or '/' not in repo_id:
# log.warning(f'Auth: repo="{repo_id}" invalid repo id')
return False
auth_ok = False
try:
login = modelloader.hf_login()
hf.auth_check(repo_id, write=False)
token = os.environ.get('HF_TOKEN', None)
hf.auth_check(repo_id, write=False, token=token)
auth_ok = True
except Exception as e:
log.error(f'Auth: repo="{repo_id}" login={login} auth={auth_ok} {e}')