chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: log issues with websocket

Browse Source
This commit is contained in:
bennykok 2024-07-22 13:36:39 -07:00
parent b4273b1907
commit 4f1a80fb64
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
custom_routes.py
View File

@ -73,11 +73,11 @@ def clear_current_prompt(sid):
prompt_server = server.PromptServer.instance prompt_server = server.PromptServer.instance
to_delete = list(streaming_prompt_metadata[sid].running_prompt_ids) # Convert set to list to_delete = list(streaming_prompt_metadata[sid].running_prompt_ids) # Convert set to list
logger.info("clearning out prompt: ", to_delete) logger.info(f"clearing out prompt: {to_delete}")
for id_to_delete in to_delete: for id_to_delete in to_delete:
delete_func = lambda a: a[1] == id_to_delete delete_func = lambda a: a[1] == id_to_delete
prompt_server.prompt_queue.delete_queue_item(delete_func) prompt_server.prompt_queue.delete_queue_item(delete_func)
logger.info("deleted prompt: ", id_to_delete, prompt_server.prompt_queue.get_tasks_remaining()) logger.info(f"deleted prompt: {id_to_delete}, remaining tasks: {prompt_server.prompt_queue.get_tasks_remaining()}")
streaming_prompt_metadata[sid].running_prompt_ids.clear() streaming_prompt_metadata[sid].running_prompt_ids.clear()