Add strength control for vace. (#7717)

comfy/ldm/wan/model.py

@@ -582,6 +582,7 @@ class VaceWanModel(WanModel):
         t,
         context,
         vace_context,
+        vace_strength=1.0,
         clip_fea=None,
         freqs=None,
         transformer_options={},
@@ -629,7 +630,7 @@ class VaceWanModel(WanModel):
             ii = self.vace_layers_mapping.get(i, None)
             if ii is not None:
                 c_skip, c = self.vace_blocks[ii](c, x=x_orig, e=e0, freqs=freqs, context=context, context_img_len=context_img_len)
-            x += c_skip
+            x += c_skip * vace_strength
         # head
         x = self.head(x, e)
 

comfy/model_base.py

@@ -1068,6 +1068,9 @@ class WAN21_Vace(WAN21):
             mask = torch.ones(noise_shape, device=noise.device, dtype=noise.dtype)
 
         out['vace_context'] = comfy.conds.CONDRegular(torch.cat([vace_frames.to(noise), mask.to(noise)], dim=1))
+
+        vace_strength = kwargs.get("vace_strength", 1.0)
+        out['vace_strength'] = comfy.conds.CONDConstant(vace_strength)
         return out