improved: When user do Switch ComfyUI, update the policy accordingly.

This commit is contained in:
Dr.Lt.Data
2025-02-20 12:19:50 +09:00
parent b0daf81185
commit e9df78c0e7
2 changed files with 17 additions and 5 deletions

View File

@@ -3264,7 +3264,9 @@ def switch_comfyui(tag):
if tag == 'nightly':
repo.git.checkout('master')
repo.remotes.origin.pull()
tracking_branch = repo.active_branch.tracking_branch()
remote_name = tracking_branch.remote_name
repo.remotes[remote_name].pull()
print("[ComfyUI-Manager] ComfyUI version is switched to the latest 'master' version")
else:
repo.git.checkout(tag)