From 7f9c134362ae35b33e5b853dcf9b96e921abc582 Mon Sep 17 00:00:00 2001 From: Diego Martin Arroyo Date: Wed, 21 Jan 2026 13:19:13 +0100 Subject: [PATCH] Fixes mismatched number of output arguments In #306 the return of WanTimeTextImageEmbedding was modified to include a 5th element. This refactors the call in WanVACEModel to comply with the new interface --- src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py b/src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py index ce73ac5d..fc3e67e3 100644 --- a/src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py +++ b/src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py @@ -474,6 +474,7 @@ def __call__( timestep_proj, encoder_hidden_states, encoder_hidden_states_image, + _, ) = self.condition_embedder( # We will need to mask out the text embedding. timestep, encoder_hidden_states, encoder_hidden_states_image )