comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-19 12:09:12 +00:00
Code Issues 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
1 changed files with 2 additions and 1 deletions
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)