print raw download url for convenience (#1478)

* print raw download url for convenience

* print raw download url for convenience
This commit is contained in:
HuangYongliang 2025-01-26 17:09:03 +08:00 committed by GitHub
parent 41998565db
commit 9a5c7c10de
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions

View File

@ -2,7 +2,7 @@ import os
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:

View File

@ -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)