From 555c67a8cd2d096fbab86d9e67d6cca6e5eb08d4 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 14 May 2025 17:33:01 +0300
Subject: [PATCH] Support 14B VACE

---
 nodes.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/nodes.py b/nodes.py
index 06a2ed7..e5e70c1 100644
--- a/nodes.py
+++ b/nodes.py
@@ -414,7 +414,6 @@ class WanVideoVACEModelSelect:
         return {
             "required": {
                 "vace_model": (folder_paths.get_filename_list("diffusion_models"), {"tooltip": "These models are loaded from the 'ComfyUI/models/diffusion_models' VACE model to use when not using model that has it included"}),
-                "vace_blocks": ("STRING", {"default": "0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28", "multiline": True, "tooltip": "Blocks to apply VACE to, default is for 1.3B model"}),
             },
         }
 
@@ -424,10 +423,9 @@ class WanVideoVACEModelSelect:
     CATEGORY = "WanVideoWrapper"
     DESCRIPTION = "VACE model to use when not using model that has it included, loaded from 'ComfyUI/models/diffusion_models'"
 
-    def getvacepath(self, vace_model, vace_blocks):
+    def getvacepath(self, vace_model):
         vace_model = {
             "path": folder_paths.get_full_path("diffusion_models", vace_model),
-            "blocks": [int(x.strip()) for x in vace_blocks.split(",")],
         }
         return (vace_model,)
 
@@ -570,8 +568,8 @@ class WanVideoModelLoader:
         vace_layers, vace_in_dim = None, None
         if "vace_blocks.0.after_proj.weight" in sd:
             model_type = "t2v"
-            if vace_model is not None:
-                vace_layers = vace_model["blocks"]
+            if dim == 5120:
+                vace_layers = [0, 5, 10, 15, 20, 25, 30, 35]
             else:
                 vace_layers = [0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28]
             vace_in_dim = 96