From e3ea2bf392213d16c2ba56f7584354e819b72575 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 21 Apr 2025 20:01:20 +0300
Subject: [PATCH] Allow VACE with DF

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

diff --git a/skyreels/nodes.py b/skyreels/nodes.py
index 3360a41..312099d 100644
--- a/skyreels/nodes.py
+++ b/skyreels/nodes.py
@@ -233,6 +233,7 @@ class WanVideoDiffusionForcingSampler:
         
         fps_embeds = None
         if hasattr(transformer, "fps_embedding"):
+            fps = round(fps, 2)
             log.info(f"Model has fps embedding, using {fps} fps")
             fps_embeds = [fps]
             fps_embeds = [0 if i == 16 else 1 for i in fps_embeds]
@@ -371,7 +372,7 @@ class WanVideoDiffusionForcingSampler:
 
         #region model pred
         def predict_with_cfg(z, cfg_scale, positive_embeds, negative_embeds, timestep, idx, image_cond=None, clip_fea=None, 
-                             control_latents=None, vace_data=None, unianim_data=None, teacache_state=None):
+                             vace_data=None, unianim_data=None, teacache_state=None):
             with torch.autocast(device_type=mm.get_autocast_device(device), dtype=model["dtype"], enabled=True):
 
                 if use_cfg_zero_star and (idx <= zero_star_steps) and use_zero_init:
@@ -490,7 +491,7 @@ class WanVideoDiffusionForcingSampler:
                 cfg[i], 
                 text_embeds["prompt_embeds"], 
                 text_embeds["negative_prompt_embeds"], 
-                timestep, i, image_cond, clip_fea, vace_data,
+                timestep, i, image_cond, clip_fea, vace_data=vace_data,
                 teacache_state=self.teacache_state)
             
             for idx in range(valid_interval_start, valid_interval_end):