xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2025-12-10 14:34:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

Disable offload stream when torch compile. (#10961)

Browse Source
This commit is contained in:
comfyanonymous 2025-11-28 13:16:46 -08:00 committed by GitHub
parent ca7808f240
commit f55c98a89f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_management.py
View File

@ -1042,6 +1042,9 @@ def get_offload_stream(device):
if NUM_STREAMS == 0: if NUM_STREAMS == 0:
return None return None
if torch.compiler.is_compiling():
return None
if device in STREAMS: if device in STREAMS:
ss = STREAMS[device] ss = STREAMS[device]
#Sync the oldest stream in the queue with the current #Sync the oldest stream in the queue with the current