TBNilles/ComfyUI-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'feat/cnr' into feat/cnr-no-suffix

Browse Source
This commit is contained in:
Dr.Lt.Data
2024-12-21 18:28:46 +09:00
parent 9503c34d5b b9227b1570
commit cbcd2e14ce
12 changed files with 3484 additions and 2910 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
glob/manager_downloader.py
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)

9
glob/manager_server.py
View File

@@ -1400,10 +1400,11 @@ async def default_cache_update():
await asyncio.gather(a, b, c, d, e)
if not core.get_config()['skip_migration_check']:
await core.check_need_to_migrate()
else:
logging.info("[ComfyUI-Manager] Migration check is skipped...")
# NOTE: hide migration button temporarily.
# if not core.get_config()['skip_migration_check']:
# await core.check_need_to_migrate()
# else:
# logging.info("[ComfyUI-Manager] Migration check is skipped...")
threading.Thread(target=lambda: asyncio.run(default_cache_update())).start()