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)