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

Enable real-time progress notifications.

Browse Source
This commit is contained in:
doctorpangloss 2024-04-08 14:56:16 -07:00
parent dd6f7c4215
commit 37cca051b6
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/distributed/distributed_progress.py
View File

@ -25,9 +25,7 @@ def _get_name(queue_name: str, user_id: str) -> str:
class DistributedExecutorToClientProgress(ExecutorToClientProgress): class DistributedExecutorToClientProgress(ExecutorToClientProgress):
def __init__(self, rpc: RPC, queue_name: str, loop: AbstractEventLoop, receive_all_progress_notifications=False): def __init__(self, rpc: RPC, queue_name: str, loop: AbstractEventLoop, receive_all_progress_notifications=True):
if receive_all_progress_notifications:
raise NotImplementedError("this release does not yet support sending all progress notifications from the hook over the network")
self._rpc = rpc self._rpc = rpc
self._queue_name = queue_name self._queue_name = queue_name
self._loop = loop self._loop = loop