Disable offload stream when torch compile. (#10961)

2025-11-28 13:16:46 -08:00 · 2025-11-28 13:16:46 -08:00 · f55c98a89f
parent ca7808f240
commit f55c98a89f
1 changed files with 3 additions and 0 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -1042,6 +1042,9 @@ def get_offload_stream(device):
    if NUM_STREAMS == 0:
        return None

+    if torch.compiler.is_compiling():
+        return None
+
    if device in STREAMS:
        ss = STREAMS[device]
        #Sync the oldest stream in the queue with the current