wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-21 03:40:16 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Fix progress updates

Browse Source
This commit is contained in:
doctorpangloss 2024-06-19 13:40:28 -07:00
parent 68f410b8da
commit 92f435aba3
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/utils.py
View File

@ -556,6 +556,7 @@ class ProgressBar:
def __init__(self, total: float): def __init__(self, total: float):
self.total: float = total self.total: float = total
self.current: float = 0.0 self.current: float = 0.0
self.server = current_execution_context().server
def update_absolute(self, value, total=None, preview_image_or_output=None): def update_absolute(self, value, total=None, preview_image_or_output=None):
if total is not None: if total is not None:
@ -563,7 +564,7 @@ class ProgressBar:
if value > self.total: if value > self.total:
value = self.total value = self.total
self.current = value self.current = value
_progress_bar_update(self.current, self.total, preview_image_or_output) _progress_bar_update(self.current, self.total, preview_image_or_output, server=self.server)
def update(self, value): def update(self, value):
self.update_absolute(self.current + value) self.update_absolute(self.current + value)
@ -581,14 +582,15 @@ def comfy_tqdm():
""" """
_original_init = tqdm.__init__ _original_init = tqdm.__init__
_original_update = tqdm.update _original_update = tqdm.update
server = current_execution_context().server
try: try:
def __init(self, *args, **kwargs): def __init(self, *args, **kwargs):
_original_init(self, *args, **kwargs) _original_init(self, *args, **kwargs)
self._progress_bar = ProgressBar(self.total)
def __update(self, n=1): def __update(self, n=1):
assert self._progress_bar is not None
_original_update(self, n) _original_update(self, n)
_progress_bar_update(n, self.total, server=server) self._progress_bar.update(n)
tqdm.__init__ = __init tqdm.__init__ = __init
tqdm.update = __update tqdm.update = __update