From 4cebb550326361286ed0a5430cc668241d65b832 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 2 Apr 2025 18:41:39 +0300
Subject: [PATCH] more robust vace check

---
 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 1cb6d20..8600625 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 > 0:
+        if vace_blocks_to_swap > 0 and self.vace_layers is not None:
             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"):