Merge branch 'main' into feat/cnr

This commit is contained in:
Dr.Lt.Data
2024-12-21 18:11:14 +09:00
11 changed files with 3479 additions and 2906 deletions

View File

@@ -38,6 +38,8 @@ def basic_download_url(url, dest_folder, filename):
def download_url(model_url: str, model_dir: str, filename: str):
if HF_ENDPOINT:
model_url = model_url.replace('https://huggingface.co', HF_ENDPOINT)
if aria2:
return aria2_download_url(model_url, model_dir, filename)
else:
@@ -66,9 +68,6 @@ def aria2_download_url(model_url: str, model_dir: str, filename: str):
if model_dir.startswith(core.comfy_path):
model_dir = model_dir[len(core.comfy_path) :]
if HF_ENDPOINT:
model_url = model_url.replace('https://huggingface.co', HF_ENDPOINT)
download_dir = model_dir if model_dir.startswith('/') else os.path.join('/models', model_dir)
download = aria2_find_task(download_dir, filename)