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

fix parse aria2 host

Browse Source
This commit is contained in:
dishuostec 2024-07-01 12:59:26 +08:00
parent 4448e9a4f1
commit e6e0921bd9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
glob/manager_downloader.py
View File

@ -1,11 +1,14 @@
import os
from urllib.parse import urlparse
aria2 = os.getenv('COMFYUI_MANAGER_ARIA2_SERVER')
HF_ENDPOINT = os.getenv('HF_ENDPOINT')
if aria2 is not None:
secret = os.getenv('COMFYUI_MANAGER_ARIA2_SECRET')
host, port = aria2.split(':')
url = urlparse(aria2)
port = url.port
host = url.scheme + '://' + url.hostname
import aria2p
aria2 = aria2p.API(aria2p.Client(host=host, port=port, secret=secret))