From 5a50c3c7e59a867ddf0b3b7cc207c877a7b422fb Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Mon, 28 Apr 2025 10:07:21 -0700 Subject: [PATCH] Fix stream priority to support older pytorch. (#7856) --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 516b6e2f..78317af3 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -962,7 +962,7 @@ def get_offload_stream(device): elif is_device_cuda(device): ss = [] for k in range(NUM_STREAMS): - ss.append(torch.cuda.Stream(device=device, priority=10)) + ss.append(torch.cuda.Stream(device=device, priority=0)) STREAMS[device] = ss s = ss[stream_counter] stream_counter = (stream_counter + 1) % len(ss)