Files
hao-ai-lab-FastVideo/fastvideo
Enderfga fba7170ed0 [feat] Thread optional r_timestep through WanTransformer3DModel.forward
The forward signature now accepts r_timestep as an explicit kwarg (not
silently swallowed by **kwargs), gets flattened in parallel with timestep
when the input is 2-D (Wan 2.2 ti2v case), and is forwarded to the
condition_embedder.

The constructor now propagates the four arch_config flags
(r_embedder/r_embedder_fusion/r_embedder_gate_value/r_embedder_deltatime_type)
into the WanTimeTextImageEmbedding so a YAML can opt into AnyFlow's
dual-timestep conditioning through training.dit_config overrides.

Defaults preserve byte-identity with the legacy single-timestep path —
when r_timestep is omitted the delta_embedder branch is skipped entirely
(test_embedder_enabled_without_r_timestep_is_bit_identical_to_legacy).
2026-07-11 18:36:14 -07:00
..
2026-06-09 14:55:55 -07:00
2026-06-09 14:55:55 -07:00
2026-06-04 14:22:49 -07:00