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"):