From 390bc2a6cb2bd9be7b15d3e8d43a96e31a03262b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 2 Apr 2025 18:38:16 +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 8a211fb..1cb6d20 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -817,7 +817,7 @@ class WanModel(ModelMixin, ConfigMixin):
                 block.to(self.offload_device, non_blocking=self.use_non_blocking)
                 total_offload_memory += block_memory
 
-        if vace_blocks_to_swap is not None:
+        if vace_blocks_to_swap > 0:
             self.vace_blocks_to_swap = vace_blocks_to_swap
 
             for b, block in tqdm(enumerate(self.vace_blocks), total=len(self.vace_blocks), desc="Initializing vace block swap"):