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

fix logging

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-04-14 08:28:35 -04:00
parent 5c3ab983c0
commit 06f4879be9
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/modelloader.py
+6 -6
View File
@@ -10,7 +10,7 @@ from urllib.parse import urlparse
from PIL import Image
import rich.progress as p
import huggingface_hub as hf
from installer import install
from installer import install, log
from modules import shared, errors, files_cache
from modules.upscaler import Upscaler
from modules.paths import script_path, models_path
@@ -18,7 +18,7 @@ from modules.paths import script_path, models_path
loggedin = None
diffuser_repos = []
debug = shared.log.trace if os.environ.get('SD_DOWNLOAD_DEBUG', None) is not None else lambda *args, **kwargs: None
debug = log.trace if os.environ.get('SD_DOWNLOAD_DEBUG', None) is not None else lambda *args, **kwargs: None
pbar = None
@@ -27,13 +27,13 @@ def hf_login(token=None):
token = token or shared.opts.huggingface_token
install('hf_xet', quiet=True)
if token is None or len(token) <= 2:
shared.log.debug('HF login: no token provided')
log.debug('HF login: no token provided')
return False
if os.environ.get('HUGGING_FACE_HUB_TOKEN', None) is not None:
shared.log.warning('HF login: removing existing env variable: HUGGING_FACE_HUB_TOKEN')
log.warning('HF login: removing existing env variable: HUGGING_FACE_HUB_TOKEN')
del os.environ['HUGGING_FACE_HUB_TOKEN']
if os.environ.get('HF_TOKEN', None) is not None:
shared.log.warning('HF login: removing existing env variable: HF_TOKEN')
log.warning('HF login: removing existing env variable: HF_TOKEN')
del os.environ['HF_TOKEN']
if loggedin != token:
stdout = io.StringIO()
@@ -42,7 +42,7 @@ def hf_login(token=None):
hf.login(token=token, add_to_git_credential=False, write_permission=False)
text = stdout.getvalue() or ''
line = [l for l in text.split('\n') if 'Token' in l]
shared.log.info(f'HF login: token="{hf.constants.HF_TOKEN_PATH}" {line[0] if len(line) > 0 else text}')
log.info(f'HF login: token="{hf.constants.HF_TOKEN_PATH}" {line[0] if len(line) > 0 else text}')
loggedin = token
return True