Update model.py

2025-08-30 23:34:14 +03:00
parent 40308f1c70
commit 10d3a0fbef
1 changed files with 1 additions and 1 deletions
@@ -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)