xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-10 10:53:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update websockets_api_example.py

Browse Source
This commit is contained in:
RandomGitUser321 2024-09-21 22:53:38 -04:00 committed by GitHub
parent 364b69e931
commit eb8b496b96
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
script_examples/websockets_api_example.py
View File

@ -38,6 +38,12 @@ def get_images(ws, prompt):
if data['node'] is None and data['prompt_id'] == prompt_id: if data['node'] is None and data['prompt_id'] == prompt_id:
break #Execution is done break #Execution is done
else: else:
# If you want to be able to decode the binary stream for latent previews, here is how you can do it:
# event = struct.unpack(">I", out[:4])[0]
# image_data = out[8:]
# if event == 1:
# bytesIO = BytesIO(image_data)
# preview_image = Image.open(bytesIO) # This is your preview in PIL image format, store it in a global
continue #previews are binary data continue #previews are binary data
history = get_history(prompt_id)[prompt_id] history = get_history(prompt_id)[prompt_id]