comfy: more changes for fast previews

This commit is contained in:
Kasisnu
2024-08-06 17:50:19 +05:30
parent 044057817a
commit 1df60fdfed
2 changed files with 29 additions and 19 deletions
@@ -221,9 +221,8 @@ async def smart_wait(ws, prompt_id):
out = await ws.recv()
if isinstance(out, str):
message = json.loads(out)
if message['type'] != 'progress':
print(message)
if message['type'] == 'execution_error':
print(message)
classify_error_message(message['data'])
if message['type'] == 'executing':
data = message['data']