From 10d3a0fbef2dd2865e6aa0fbf64201bf484ea18b Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sat, 30 Aug 2025 23:34:14 +0300 Subject: [PATCH] Update model.py --- wanvideo/modules/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py index fcc52b7..a27041b 100644 --- a/wanvideo/modules/model.py +++ b/wanvideo/modules/model.py @@ -2520,7 +2520,7 @@ class WanModel(torch.nn.Module): self.controlnet.to(self.offload_device) # Asynchronous block offloading with CUDA streams and events - cuda_stream = torch.cuda.Stream(device=device, priority=0) + cuda_stream = None #torch.cuda.Stream(device=device, priority=0) events = [torch.cuda.Event() for _ in self.blocks] swap_start_idx = len(self.blocks) - self.blocks_to_swap if self.blocks_to_swap > 0 else len(self.blocks)