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

feat: config.ini - model_download_by_agent

Browse Source 
https://github.com/ltdrdata/ComfyUI-Manager/issues/156
This commit is contained in:
dr.lt.data 2024-03-04 16:52:56 +09:00
parent a04ba7f01b
commit 7f95b824c5
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
__init__.py
View File

@ -29,7 +29,7 @@ except:
print(f"[WARN] ComfyUI-Manager: Your ComfyUI version is outdated. Please update to the latest version.") print(f"[WARN] ComfyUI-Manager: Your ComfyUI version is outdated. Please update to the latest version.")
version = [2, 8, 3] version = [2, 9]
version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '') version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '')
print(f"### Loading: ComfyUI-Manager ({version_str})") print(f"### Loading: ComfyUI-Manager ({version_str})")
@ -178,6 +178,7 @@ def write_config():
'component_policy': get_config()['component_policy'], 'component_policy': get_config()['component_policy'],
'double_click_policy': get_config()['double_click_policy'], 'double_click_policy': get_config()['double_click_policy'],
'windows_selector_event_loop_policy': get_config()['windows_selector_event_loop_policy'], 'windows_selector_event_loop_policy': get_config()['windows_selector_event_loop_policy'],
'model_download_by_agent': get_config()['model_download_by_agent']
} }
with open(config_path, 'w') as configfile: with open(config_path, 'w') as configfile:
config.write(configfile) config.write(configfile)
@ -201,6 +202,7 @@ def read_config():
'component_policy': default_conf['component_policy'] if 'component_policy' in default_conf else 'workflow', 'component_policy': default_conf['component_policy'] if 'component_policy' in default_conf else 'workflow',
'double_click_policy': default_conf['double_click_policy'] if 'double_click_policy' in default_conf else 'copy-all', 'double_click_policy': default_conf['double_click_policy'] if 'double_click_policy' in default_conf else 'copy-all',
'windows_selector_event_loop_policy': default_conf['windows_selector_event_loop_policy'] if 'windows_selector_event_loop_policy' in default_conf else False, 'windows_selector_event_loop_policy': default_conf['windows_selector_event_loop_policy'] if 'windows_selector_event_loop_policy' in default_conf else False,
'model_download_by_agent': default_conf['model_download_by_agent'] if 'model_download_by_agent' in default_conf else False,
} }
except Exception: except Exception:
@ -215,7 +217,8 @@ def read_config():
'default_ui': 'none', 'default_ui': 'none',
'component_policy': 'workflow', 'component_policy': 'workflow',
'double_click_policy': 'copy-all', 'double_click_policy': 'copy-all',
'windows_selector_event_loop_policy': False 'windows_selector_event_loop_policy': False,
'model_download_by_agent': False,
} }
@ -1755,8 +1758,7 @@ async def install_model(request):
print(f"Install model '{json_data['name']}' into '{model_path}'") print(f"Install model '{json_data['name']}' into '{model_path}'")
model_url = json_data['url'] model_url = json_data['url']
if not get_config()['model_download_by_agent'] and (model_url.startswith('https://github.com') or model_url.startswith('https://huggingface.co') or model_url.startswith('https://heibox.uni-heidelberg.de')):
if model_url.startswith('https://github.com') or model_url.startswith('https://huggingface.co') or model_url.startswith('https://heibox.uni-heidelberg.de'):
model_dir = get_model_dir(json_data) model_dir = get_model_dir(json_data)
download_url(model_url, model_dir, filename=json_data['filename']) download_url(model_url, model_dir, filename=json_data['filename'])