From 37cca051b68620cd8101e3f3dc22a90292810e53 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Mon, 8 Apr 2024 14:56:16 -0700
Subject: [PATCH] Enable real-time progress notifications.

---
 comfy/distributed/distributed_progress.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/comfy/distributed/distributed_progress.py b/comfy/distributed/distributed_progress.py
index 11f73fcfe..6e971994d 100644
--- a/comfy/distributed/distributed_progress.py
+++ b/comfy/distributed/distributed_progress.py
@@ -25,9 +25,7 @@ def _get_name(queue_name: str, user_id: str) -> str:
 
 
 class DistributedExecutorToClientProgress(ExecutorToClientProgress):
-    def __init__(self, rpc: RPC, queue_name: str, loop: AbstractEventLoop, receive_all_progress_notifications=False):
-        if receive_all_progress_notifications:
-            raise NotImplementedError("this release does not yet support sending all progress notifications from the hook over the network")
+    def __init__(self, rpc: RPC, queue_name: str, loop: AbstractEventLoop, receive_all_progress_notifications=True):
         self._rpc = rpc
         self._queue_name = queue_name
         self._loop = loop