Spaces:
Running
on
Zero
Running
on
Zero
Upload 8 files
Browse files- .ci/update_windows/update.py +146 -0
- .ci/update_windows/update_comfyui.bat +8 -0
- .ci/update_windows/update_comfyui_stable.bat +8 -0
- .ci/windows_base_files/README_VERY_IMPORTANT.txt +31 -0
- .ci/windows_base_files/run_cpu.bat +2 -0
- .ci/windows_base_files/run_nvidia_gpu.bat +2 -0
- .ci/windows_nightly_base_files/run_nvidia_gpu_fast.bat +2 -0
- .ci/windows_nightly_base_files/run_nvidia_gpu_fast_fp16_accumulation.bat +2 -0
.ci/update_windows/update.py
ADDED
@@ -0,0 +1,146 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import pygit2
|
2 |
+
from datetime import datetime
|
3 |
+
import sys
|
4 |
+
import os
|
5 |
+
import shutil
|
6 |
+
import filecmp
|
7 |
+
|
8 |
+
def pull(repo, remote_name='origin', branch='master'):
|
9 |
+
for remote in repo.remotes:
|
10 |
+
if remote.name == remote_name:
|
11 |
+
remote.fetch()
|
12 |
+
remote_master_id = repo.lookup_reference('refs/remotes/origin/%s' % (branch)).target
|
13 |
+
merge_result, _ = repo.merge_analysis(remote_master_id)
|
14 |
+
# Up to date, do nothing
|
15 |
+
if merge_result & pygit2.GIT_MERGE_ANALYSIS_UP_TO_DATE:
|
16 |
+
return
|
17 |
+
# We can just fastforward
|
18 |
+
elif merge_result & pygit2.GIT_MERGE_ANALYSIS_FASTFORWARD:
|
19 |
+
repo.checkout_tree(repo.get(remote_master_id))
|
20 |
+
try:
|
21 |
+
master_ref = repo.lookup_reference('refs/heads/%s' % (branch))
|
22 |
+
master_ref.set_target(remote_master_id)
|
23 |
+
except KeyError:
|
24 |
+
repo.create_branch(branch, repo.get(remote_master_id))
|
25 |
+
repo.head.set_target(remote_master_id)
|
26 |
+
elif merge_result & pygit2.GIT_MERGE_ANALYSIS_NORMAL:
|
27 |
+
repo.merge(remote_master_id)
|
28 |
+
|
29 |
+
if repo.index.conflicts is not None:
|
30 |
+
for conflict in repo.index.conflicts:
|
31 |
+
print('Conflicts found in:', conflict[0].path) # noqa: T201
|
32 |
+
raise AssertionError('Conflicts, ahhhhh!!')
|
33 |
+
|
34 |
+
user = repo.default_signature
|
35 |
+
tree = repo.index.write_tree()
|
36 |
+
repo.create_commit('HEAD',
|
37 |
+
user,
|
38 |
+
user,
|
39 |
+
'Merge!',
|
40 |
+
tree,
|
41 |
+
[repo.head.target, remote_master_id])
|
42 |
+
# We need to do this or git CLI will think we are still merging.
|
43 |
+
repo.state_cleanup()
|
44 |
+
else:
|
45 |
+
raise AssertionError('Unknown merge analysis result')
|
46 |
+
|
47 |
+
pygit2.option(pygit2.GIT_OPT_SET_OWNER_VALIDATION, 0)
|
48 |
+
repo_path = str(sys.argv[1])
|
49 |
+
repo = pygit2.Repository(repo_path)
|
50 |
+
ident = pygit2.Signature('comfyui', 'comfy@ui')
|
51 |
+
try:
|
52 |
+
print("stashing current changes") # noqa: T201
|
53 |
+
repo.stash(ident)
|
54 |
+
except KeyError:
|
55 |
+
print("nothing to stash") # noqa: T201
|
56 |
+
backup_branch_name = 'backup_branch_{}'.format(datetime.today().strftime('%Y-%m-%d_%H_%M_%S'))
|
57 |
+
print("creating backup branch: {}".format(backup_branch_name)) # noqa: T201
|
58 |
+
try:
|
59 |
+
repo.branches.local.create(backup_branch_name, repo.head.peel())
|
60 |
+
except:
|
61 |
+
pass
|
62 |
+
|
63 |
+
print("checking out master branch") # noqa: T201
|
64 |
+
branch = repo.lookup_branch('master')
|
65 |
+
if branch is None:
|
66 |
+
ref = repo.lookup_reference('refs/remotes/origin/master')
|
67 |
+
repo.checkout(ref)
|
68 |
+
branch = repo.lookup_branch('master')
|
69 |
+
if branch is None:
|
70 |
+
repo.create_branch('master', repo.get(ref.target))
|
71 |
+
else:
|
72 |
+
ref = repo.lookup_reference(branch.name)
|
73 |
+
repo.checkout(ref)
|
74 |
+
|
75 |
+
print("pulling latest changes") # noqa: T201
|
76 |
+
pull(repo)
|
77 |
+
|
78 |
+
if "--stable" in sys.argv:
|
79 |
+
def latest_tag(repo):
|
80 |
+
versions = []
|
81 |
+
for k in repo.references:
|
82 |
+
try:
|
83 |
+
prefix = "refs/tags/v"
|
84 |
+
if k.startswith(prefix):
|
85 |
+
version = list(map(int, k[len(prefix):].split(".")))
|
86 |
+
versions.append((version[0] * 10000000000 + version[1] * 100000 + version[2], k))
|
87 |
+
except:
|
88 |
+
pass
|
89 |
+
versions.sort()
|
90 |
+
if len(versions) > 0:
|
91 |
+
return versions[-1][1]
|
92 |
+
return None
|
93 |
+
latest_tag = latest_tag(repo)
|
94 |
+
if latest_tag is not None:
|
95 |
+
repo.checkout(latest_tag)
|
96 |
+
|
97 |
+
print("Done!") # noqa: T201
|
98 |
+
|
99 |
+
self_update = True
|
100 |
+
if len(sys.argv) > 2:
|
101 |
+
self_update = '--skip_self_update' not in sys.argv
|
102 |
+
|
103 |
+
update_py_path = os.path.realpath(__file__)
|
104 |
+
repo_update_py_path = os.path.join(repo_path, ".ci/update_windows/update.py")
|
105 |
+
|
106 |
+
cur_path = os.path.dirname(update_py_path)
|
107 |
+
|
108 |
+
|
109 |
+
req_path = os.path.join(cur_path, "current_requirements.txt")
|
110 |
+
repo_req_path = os.path.join(repo_path, "requirements.txt")
|
111 |
+
|
112 |
+
|
113 |
+
def files_equal(file1, file2):
|
114 |
+
try:
|
115 |
+
return filecmp.cmp(file1, file2, shallow=False)
|
116 |
+
except:
|
117 |
+
return False
|
118 |
+
|
119 |
+
def file_size(f):
|
120 |
+
try:
|
121 |
+
return os.path.getsize(f)
|
122 |
+
except:
|
123 |
+
return 0
|
124 |
+
|
125 |
+
|
126 |
+
if self_update and not files_equal(update_py_path, repo_update_py_path) and file_size(repo_update_py_path) > 10:
|
127 |
+
shutil.copy(repo_update_py_path, os.path.join(cur_path, "update_new.py"))
|
128 |
+
exit()
|
129 |
+
|
130 |
+
if not os.path.exists(req_path) or not files_equal(repo_req_path, req_path):
|
131 |
+
import subprocess
|
132 |
+
try:
|
133 |
+
subprocess.check_call([sys.executable, '-s', '-m', 'pip', 'install', '-r', repo_req_path])
|
134 |
+
shutil.copy(repo_req_path, req_path)
|
135 |
+
except:
|
136 |
+
pass
|
137 |
+
|
138 |
+
|
139 |
+
stable_update_script = os.path.join(repo_path, ".ci/update_windows/update_comfyui_stable.bat")
|
140 |
+
stable_update_script_to = os.path.join(cur_path, "update_comfyui_stable.bat")
|
141 |
+
|
142 |
+
try:
|
143 |
+
if not file_size(stable_update_script_to) > 10:
|
144 |
+
shutil.copy(stable_update_script, stable_update_script_to)
|
145 |
+
except:
|
146 |
+
pass
|
.ci/update_windows/update_comfyui.bat
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
@echo off
|
2 |
+
..\python_embeded\python.exe .\update.py ..\ComfyUI\
|
3 |
+
if exist update_new.py (
|
4 |
+
move /y update_new.py update.py
|
5 |
+
echo Running updater again since it got updated.
|
6 |
+
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --skip_self_update
|
7 |
+
)
|
8 |
+
if "%~1"=="" pause
|
.ci/update_windows/update_comfyui_stable.bat
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
@echo off
|
2 |
+
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --stable
|
3 |
+
if exist update_new.py (
|
4 |
+
move /y update_new.py update.py
|
5 |
+
echo Running updater again since it got updated.
|
6 |
+
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --skip_self_update --stable
|
7 |
+
)
|
8 |
+
if "%~1"=="" pause
|
.ci/windows_base_files/README_VERY_IMPORTANT.txt
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
HOW TO RUN:
|
2 |
+
|
3 |
+
if you have a NVIDIA gpu:
|
4 |
+
|
5 |
+
run_nvidia_gpu.bat
|
6 |
+
|
7 |
+
|
8 |
+
|
9 |
+
To run it in slow CPU mode:
|
10 |
+
|
11 |
+
run_cpu.bat
|
12 |
+
|
13 |
+
|
14 |
+
|
15 |
+
IF YOU GET A RED ERROR IN THE UI MAKE SURE YOU HAVE A MODEL/CHECKPOINT IN: ComfyUI\models\checkpoints
|
16 |
+
|
17 |
+
You can download the stable diffusion 1.5 one from: https://huggingface.co/Comfy-Org/stable-diffusion-v1-5-archive/blob/main/v1-5-pruned-emaonly-fp16.safetensors
|
18 |
+
|
19 |
+
|
20 |
+
RECOMMENDED WAY TO UPDATE:
|
21 |
+
To update the ComfyUI code: update\update_comfyui.bat
|
22 |
+
|
23 |
+
|
24 |
+
|
25 |
+
To update ComfyUI with the python dependencies, note that you should ONLY run this if you have issues with python dependencies.
|
26 |
+
update\update_comfyui_and_python_dependencies.bat
|
27 |
+
|
28 |
+
|
29 |
+
TO SHARE MODELS BETWEEN COMFYUI AND ANOTHER UI:
|
30 |
+
In the ComfyUI directory you will find a file: extra_model_paths.yaml.example
|
31 |
+
Rename this file to: extra_model_paths.yaml and edit it with your favorite text editor.
|
.ci/windows_base_files/run_cpu.bat
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
.\python_embeded\python.exe -s ComfyUI\main.py --cpu --windows-standalone-build
|
2 |
+
pause
|
.ci/windows_base_files/run_nvidia_gpu.bat
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
.\python_embeded\python.exe -s ComfyUI\main.py --windows-standalone-build
|
2 |
+
pause
|
.ci/windows_nightly_base_files/run_nvidia_gpu_fast.bat
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
.\python_embeded\python.exe -s ComfyUI\main.py --windows-standalone-build --fast
|
2 |
+
pause
|
.ci/windows_nightly_base_files/run_nvidia_gpu_fast_fp16_accumulation.bat
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
.\python_embeded\python.exe -s ComfyUI\main.py --windows-standalone-build --fast fp16_accumulation
|
2 |
+
pause
|