diff --git a/glob/manager_downloader.py b/glob/manager_downloader.py index b6393b26..41ba7649 100644 --- a/glob/manager_downloader.py +++ b/glob/manager_downloader.py @@ -2,7 +2,7 @@ from urllib.parse import urlparse import urllib import sys - +import logging aria2 = os.getenv('COMFYUI_MANAGER_ARIA2_SERVER') HF_ENDPOINT = os.getenv('HF_ENDPOINT') @@ -44,6 +44,7 @@ def basic_download_url(url, dest_folder: str, filename: str): def download_url(model_url: str, model_dir: str, filename: str): if HF_ENDPOINT: model_url = model_url.replace('https://huggingface.co', HF_ENDPOINT) + logging.info(f"model_url replaced by HF_ENDPOINT, new = {model_url}") if aria2: return aria2_download_url(model_url, model_dir, filename) else: diff --git a/glob/manager_server.py b/glob/manager_server.py index a686f9c4..e9f29a3b 100644 --- a/glob/manager_server.py +++ b/glob/manager_server.py @@ -1143,9 +1143,9 @@ async def install_model(request): try: if model_path is not None: - logging.info(f"Install model '{json_data['name']}' into '{model_path}'") model_url = json_data['url'] + logging.info(f"Install model '{json_data['name']}' from '{model_url}' into '{model_path}'") if not core.get_config()['model_download_by_agent'] and ( model_url.startswith('https://github.com') or model_url.startswith('https://huggingface.co') or model_url.startswith('https://heibox.uni-heidelberg.de')): model_dir = get_model_dir(json_data, True)