diff --git a/nodes.py b/nodes.py
index 5c58d93..1a76afa 100644
--- a/nodes.py
+++ b/nodes.py
@@ -2419,6 +2419,8 @@ class WanVideoSampler:
             has_ref = image_embeds.get("has_ref", False)
             vace_context = image_embeds.get("vace_context", None)
             vace_scale = image_embeds.get("vace_scale", None)
+            if not isinstance(vace_scale, list):
+                vace_scale = [vace_scale] * (steps+1)
             vace_start_percent = image_embeds.get("vace_start_percent", 0.0)
             vace_end_percent = image_embeds.get("vace_end_percent", 1.0)
             vace_seqlen = image_embeds.get("vace_seq_len", None)
@@ -2437,9 +2439,12 @@ class WanVideoSampler:
                     for i in range(len(vace_additional_embeds)):
                         if vace_additional_embeds[i].get("has_ref", False):
                             has_ref = True
+                        vace_scale = vace_additional_embeds[i]["vace_scale"]
+                        if not isinstance(vace_scale, list):
+                            vace_scale = [vace_scale] * (steps+1)
                         vace_data.append({
                             "context": vace_additional_embeds[i]["vace_context"],
-                            "scale": vace_additional_embeds[i]["vace_scale"],
+                            "scale": vace_scale,
                             "start": vace_additional_embeds[i]["vace_start_percent"],
                             "end": vace_additional_embeds[i]["vace_end_percent"],
                             "seq_len": vace_additional_embeds[i]["vace_seq_len"]
diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index e0b2210..3dd1710 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -1405,7 +1405,7 @@ class WanModel(ModelMixin, ConfigMixin):
 
                             vace_hints = self.forward_vace(x, data["context"], data["seq_len"], kwargs)
                             vace_hint_list.append(vace_hints)
-                            vace_scale_list.append(data["scale"])
+                            vace_scale_list.append(data["scale"][current_step])
                 else:
                     vace_hints = self.forward_vace(x, vace_data, seq_len, kwargs)
                     vace_hint_list.append(vace_hints)