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

fix: fallback when ssl error on model downloading

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-11-02 18:26:08 +09:00
parent 466b8a5ef8
commit ea0c38b846
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
__init__.py
View File

@ -56,7 +56,7 @@ sys.path.append('../..')
from torchvision.datasets.utils import download_url from torchvision.datasets.utils import download_url
# ensure .js # ensure .js
print("### Loading: ComfyUI-Manager (V0.36)") print("### Loading: ComfyUI-Manager (V0.36.1)")
comfy_ui_required_revision = 1240 comfy_ui_required_revision = 1240
comfy_ui_revision = "Unknown" comfy_ui_revision = "Unknown"
@ -1132,7 +1132,13 @@ async def install_model(request):
if json_data['url'].startswith('https://github.com') or json_data['url'].startswith('https://huggingface.co'): if json_data['url'].startswith('https://github.com') or json_data['url'].startswith('https://huggingface.co'):
model_dir = get_model_dir(json_data) model_dir = get_model_dir(json_data)
download_url(json_data['url'], model_dir)
try:
download_url(json_data['url'], model_dir)
except:
fallback_url = json_data['url'].replace('https', 'http')
download_url(fallback_url, model_dir)
return web.json_response({}, content_type='application/json') return web.json_response({}, content_type='application/json')
else: else:
res = download_url_with_agent(json_data['url'], model_path) res = download_url_with_agent(json_data['url'], model_path)