print raw download url for convenience (#1478)

* print raw download url for convenience * print raw download url for convenience
ltdrdata · Jan 26, 2025 · 9a5c7c1 · 9a5c7c1
1 parent 4199856
commit 9a5c7c1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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)