nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Support list input for VACE strength

Browse Source 
Allows finer control of VACE strength over timesteps
This commit is contained in:
kijai
2025-05-21 19:04:25 +03:00
parent fa7015315a
commit 7875031efe
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+6 -1
View File
@@ -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"]
wanvideo/modules/model.py
+1 -1
View File
@@ -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)