wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-08 00:02:31 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

prefetch: guard against no offload (#13703)

Browse Source 
cast_ will return no stream if there is no work to do. guard against
this is the consume logic.
This commit is contained in:
rattus 2026-05-05 05:56:05 +10:00 committed by GitHub
parent c33d26c283
commit c47633f3be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/model_prefetch.py
View File

@ -37,6 +37,7 @@ def prefetch_queue_pop(queue, device, module):
consumed = queue.pop(0) consumed = queue.pop(0)
if consumed is not None: if consumed is not None:
offload_stream, prefetch_state = consumed offload_stream, prefetch_state = consumed
if offload_stream is not None:
offload_stream.wait_stream(comfy.model_management.current_stream(device)) offload_stream.wait_stream(comfy.model_management.current_stream(device))
_, comfy_modules = prefetch_state _, comfy_modules = prefetch_state
if comfy_modules is not None: if comfy_modules is not None: