Files
hao-ai-lab-FastVideo/fastvideo/models
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
..
2025-07-24 22:43:12 -07:00
2025-12-21 00:57:52 -06:00
2025-07-24 22:43:12 -07:00