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

feat: support better reboot mechanism based on comfy-cli

Browse Source
This commit is contained in:
Dr.Lt.Data 2024-04-23 20:46:54 +09:00
parent 4c8635d893
commit 12a364dd6a
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
glob/manager_core.py
View File

@ -21,7 +21,7 @@ sys.path.append(glob_path)
import cm_global import cm_global
from manager_util import * from manager_util import *
version = [2, 21, 3] version = [2, 22]
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 '')
comfyui_manager_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) comfyui_manager_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))

8
glob/manager_server.py
View File

@ -1112,6 +1112,14 @@ def restart(self):
except Exception as e: except Exception as e:
pass pass
if '__COMFY_CLI_SESSION__' in os.environ:
with open(os.path.join(os.environ['__COMFY_CLI_SESSION__'], '.reboot'), 'w') as file:
pass
print(f"\nRestarting...\n\n")
exit(0)
print(f"\nRestarting... [Legacy Mode]\n\n")
return os.execv(sys.executable, [sys.executable] + sys.argv) return os.execv(sys.executable, [sys.executable] + sys.argv)