mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-05 13:56:13 +02:00
Code Issues Actions 20 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

3
comfy/model_prefetch.py
View File

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