From df0b8c419afcb2ac79062b1ac611bd3b154da317 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 30 Aug 2025 18:01:05 +0300
Subject: [PATCH] Fix

---
 wanvideo/modules/model.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 89b85c5..d08e19e 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -2160,8 +2160,7 @@ class WanModel(torch.nn.Module):
                     if self.prefetch_blocks > 0 and events is not None:
                         if not events[b].query():
                             events[b].synchronize()
-                    else:
-                        block.to(self.main_device)
+                    block.to(self.main_device)
                 if self.block_swap_debug:
                     transfer_end = time.perf_counter()
                     transfer_time = transfer_end - transfer_start