jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Add the prompt_id to some websocket messages.

Browse Source
This commit is contained in:
comfyanonymous
2023-05-11 01:22:40 -04:00
parent 974958ff81
commit dfc74c19d9
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
execution.py
View File

@@ -147,7 +147,7 @@ class PromptExecutor:
self.old_prompt = {}
self.server = server
def execute(self, prompt, extra_data={}, execute_outputs=[]):
def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]):
nodes.interrupt_processing(False)
if "client_id" in extra_data:
@@ -170,7 +170,7 @@ class PromptExecutor:
current_outputs = set(self.outputs.keys())
if self.server.client_id is not None:
self.server.send_sync("execution_cached", { "nodes": list(current_outputs) }, self.server.client_id)
self.server.send_sync("execution_cached", { "nodes": list(current_outputs) , "prompt_id": prompt_id}, self.server.client_id)
executed = set()
try:
to_execute = []
@@ -190,7 +190,7 @@ class PromptExecutor:
message = str(traceback.format_exc())
print(message)
if self.server.client_id is not None:
self.server.send_sync("execution_error", { "message": message }, self.server.client_id)
self.server.send_sync("execution_error", { "message": message, "prompt_id": prompt_id }, self.server.client_id)
to_delete = []
for o in self.outputs:
@@ -207,7 +207,7 @@ class PromptExecutor:
self.old_prompt[x] = copy.deepcopy(prompt[x])
self.server.last_node_id = None
if self.server.client_id is not None:
self.server.send_sync("executing", { "node": None }, self.server.client_id)
self.server.send_sync("executing", { "node": None, "prompt_id": prompt_id }, self.server.client_id)
gc.collect()
comfy.model_management.soft_empty_cache()