feat: config.ini - skip_migration_check is supported.

This commit is contained in:
Dr.Lt.Data 2024-08-04 22:39:13 +09:00
parent 0b6f7962a4
commit e16e72cbbd
3 changed files with 11 additions and 1 deletions

View File

@ -320,6 +320,9 @@ NODE_CLASS_MAPPINGS.update({
* Use `aria2` as downloader
* [howto](docs/en/use_aria2.md)
* If you add the item `skip_migration_check = True` to `config.ini`, it will not check whether there are nodes that can be migrated at startup.
* This option can be used if performance issues occur in a Colab+GDrive environment.
## Scanner
When you run the `scan.sh` script:

View File

@ -1397,6 +1397,7 @@ def write_config():
'model_download_by_agent': get_config()['model_download_by_agent'],
'downgrade_blacklist': get_config()['downgrade_blacklist'],
'security_level': get_config()['security_level'],
'skip_migration_check': get_config()['skip_migration_check'],
}
with open(config_path, 'w') as configfile:
config.write(configfile)
@ -1431,6 +1432,7 @@ def read_config():
'windows_selector_event_loop_policy': default_conf['windows_selector_event_loop_policy'].lower() == 'true' if 'windows_selector_event_loop_policy' in default_conf else False,
'model_download_by_agent': default_conf['model_download_by_agent'].lower() == 'true' if 'model_download_by_agent' in default_conf else False,
'downgrade_blacklist': default_conf['downgrade_blacklist'] if 'downgrade_blacklist' in default_conf else '',
'skip_migration_check': default_conf['skip_migration_check'].lower() == 'true' if 'skip_migration_check' in default_conf else False,
'security_level': security_level
}
@ -1449,6 +1451,7 @@ def read_config():
'windows_selector_event_loop_policy': False,
'model_download_by_agent': False,
'downgrade_blacklist': '',
'skip_migration_check': False,
'security_level': 'normal',
}

View File

@ -1723,7 +1723,11 @@ async def default_cache_update():
e = get_cache("github-stats.json")
await asyncio.gather(a, b, c, d, e)
await core.check_need_to_migrate()
if not core.get_config()['skip_migration_check']:
await core.check_need_to_migrate()
else:
print("[ComfyUI-Manager] Migration check is skipped...")
threading.Thread(target=lambda: asyncio.run(default_cache_update())).start()