mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-01-11 02:15:17 +00:00
Compare commits
5 Commits
b3fc5694d7
...
8d01ace0b6
Author | SHA1 | Date | |
---|---|---|---|
|
8d01ace0b6 | ||
|
2307ff6746 | ||
|
29142227e3 | ||
|
37f38828f1 | ||
|
6a4064fd7e |
@ -5,10 +5,10 @@ import os
|
||||
import shutil
|
||||
import filecmp
|
||||
|
||||
def pull(repo, remote_name='origin', branch='master'):
|
||||
def pull(repo, remote_name='origin', branch='master', proxy=None):
|
||||
for remote in repo.remotes:
|
||||
if remote.name == remote_name:
|
||||
remote.fetch()
|
||||
remote.fetch(proxy=proxy)
|
||||
remote_master_id = repo.lookup_reference('refs/remotes/origin/%s' % (branch)).target
|
||||
merge_result, _ = repo.merge_analysis(remote_master_id)
|
||||
# Up to date, do nothing
|
||||
@ -46,6 +46,14 @@ def pull(repo, remote_name='origin', branch='master'):
|
||||
|
||||
pygit2.option(pygit2.GIT_OPT_SET_OWNER_VALIDATION, 0)
|
||||
repo_path = str(sys.argv[1])
|
||||
proxy = None
|
||||
if '--proxy' in sys.argv:
|
||||
proxy_index = sys.argv.index('--proxy')
|
||||
if proxy_index + 1 < len(sys.argv):
|
||||
proxy = sys.argv[proxy_index + 1]
|
||||
if len(proxy)<=0:
|
||||
proxy = None
|
||||
|
||||
repo = pygit2.Repository(repo_path)
|
||||
ident = pygit2.Signature('comfyui', 'comfy@ui')
|
||||
try:
|
||||
@ -73,7 +81,7 @@ else:
|
||||
repo.checkout(ref)
|
||||
|
||||
print("pulling latest changes") # noqa: T201
|
||||
pull(repo)
|
||||
pull(repo, proxy=proxy)
|
||||
|
||||
if "--stable" in sys.argv:
|
||||
def latest_tag(repo):
|
||||
|
@ -1,8 +1,10 @@
|
||||
@echo off
|
||||
..\python_embeded\python.exe .\update.py ..\ComfyUI\
|
||||
:: Set the http proxy here like `set proxy="http://127.0.0.1:888/"`. No spacebar allowed.
|
||||
set proxy=""
|
||||
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --proxy %proxy%
|
||||
if exist update_new.py (
|
||||
move /y update_new.py update.py
|
||||
echo Running updater again since it got updated.
|
||||
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --skip_self_update
|
||||
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --skip_self_update --proxy %proxy%
|
||||
)
|
||||
if "%~1"=="" pause
|
||||
|
@ -1,8 +1,10 @@
|
||||
@echo off
|
||||
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --stable
|
||||
:: Set the http proxy here like `set proxy="http://127.0.0.1:888/"`. No spacebar allowed.
|
||||
set proxy=""
|
||||
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --proxy %proxy% --stable
|
||||
if exist update_new.py (
|
||||
move /y update_new.py update.py
|
||||
echo Running updater again since it got updated.
|
||||
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --skip_self_update --stable
|
||||
..\python_embeded\python.exe .\update.py ..\ComfyUI\ --skip_self_update --proxy %proxy% --stable
|
||||
)
|
||||
if "%~1"=="" pause
|
||||
|
@ -111,7 +111,7 @@ class CLIP:
|
||||
model_management.load_models_gpu([self.patcher], force_full_load=True)
|
||||
self.layer_idx = None
|
||||
self.use_clip_schedule = False
|
||||
logging.info("CLIP model load device: {}, offload device: {}, current: {}, dtype: {}".format(load_device, offload_device, params['device'], dtype))
|
||||
logging.info("CLIP/text encoder model load device: {}, offload device: {}, current: {}, dtype: {}".format(load_device, offload_device, params['device'], dtype))
|
||||
|
||||
def clone(self):
|
||||
n = CLIP(no_init=True)
|
||||
@ -898,7 +898,7 @@ def load_state_dict_guess_config(sd, output_vae=True, output_clip=True, output_c
|
||||
if output_model:
|
||||
model_patcher = comfy.model_patcher.ModelPatcher(model, load_device=load_device, offload_device=model_management.unet_offload_device())
|
||||
if inital_load_device != torch.device("cpu"):
|
||||
logging.info("loaded straight to GPU")
|
||||
logging.info("loaded diffusion model directly to GPU")
|
||||
model_management.load_models_gpu([model_patcher], force_full_load=True)
|
||||
|
||||
return (model_patcher, clip, vae, clipvision)
|
||||
|
@ -4,7 +4,8 @@ lint.ignore = ["ALL"]
|
||||
# Enable specific rules
|
||||
lint.select = [
|
||||
"S307", # suspicious-eval-usage
|
||||
"T201", # print-usage
|
||||
"S102", # exec
|
||||
"T", # print-usage
|
||||
"W",
|
||||
# The "F" series in Ruff stands for "Pyflakes" rules, which catch various Python syntax errors and undefined names.
|
||||
# See all rules here: https://docs.astral.sh/ruff/rules/#pyflakes-f
|
||||
|
Loading…
Reference in New Issue
Block a user