more robust vace check

2025-04-02 18:41:39 +03:00
parent 390bc2a6cb
commit 4cebb55032
1 changed files with 1 additions and 1 deletions
@@ -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"):