wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2025-12-17 02:12:58 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Merge 2bcae64cd4 into de64af4a68

Browse Source
This commit is contained in:
Sky Zhang 2025-12-08 09:05:55 +08:00 committed by GitHub
commit 8bb723bce1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
glob/manager_downloader.py
View File

@ -80,10 +80,8 @@ def aria2_download_url(model_url: str, model_dir: str, filename: str):
import tqdm
import time
if model_dir.startswith(core.comfy_path):
model_dir = model_dir[len(core.comfy_path) :]
download_dir = model_dir if model_dir.startswith('/') else os.path.join('/models', model_dir)
download_dir = model_dir
filename += '.downloading' # aria2 uses .downloading suffix for incomplete files
download = aria2_find_task(download_dir, filename)
if download is None:
@ -104,6 +102,17 @@ def aria2_download_url(model_url: str, model_dir: str, filename: str):
progress_bar.update(download.completed_length - progress_bar.n)
time.sleep(1)
download.update()
if download.is_complete and download.completed_length == download.total_length:
file_path = os.path.join(download_dir, filename)
final_path = file_path[:-len('.downloading')] # remove the .downloading suffix
if os.path.exists(file_path):
os.rename(file_path, final_path)
logging.info(f"Download completed: {final_path}")
else:
logging.error(f"Download directory {file_path} does not exist after download completion.")
else:
logging.error(f"Download failed or incomplete: {download.status} for {model_url}")
def download_url_with_agent(url, save_path):