[feat] Add AnyFlow any-step video distillation (pretrain + on-policy)#1371

Merged

mergify[bot] merged 16 commits into

hao-ai-lab:mainfrom

Enderfga:add-anyflow-pretrain-onpolicy

Jul 12, 2026

Commits on Jul 12, 2026

[feat] Add r_embedder config flags to Wan (defaults preserve bit-identity)

Enderfga
authored and
SolitaryThinker
committed
[feat] Extend WanTimeTextImageEmbedding with optional r_timestep

Enderfga
authored and
SolitaryThinker
committed
[feat] Thread optional r_timestep through WanTransformer3DModel.forward

Enderfga
authored and
SolitaryThinker
committed
[feat] Add FlowMapEulerDiscreteScheduler

Enderfga
authored and
SolitaryThinker
committed
[feat] AnyFlowPretrainMethod skeleton + (t, r) per-batch sampling

Enderfga
authored and
SolitaryThinker
committed
[test] AnyFlow pretrain: cover (t, r) sampling distribution + validation

Enderfga
authored and
SolitaryThinker
committed
[feat] AnyFlow pretrain: central-difference target + full single_train_step

Enderfga
authored and
SolitaryThinker
committed
[feat] AnyFlowMethod (on-policy DMD2 with multi-step Euler-flow rollout)

Enderfga
authored and
SolitaryThinker
committed
[feat] AnyFlow YAML configs (pretrain + on-policy) for Wan 2.1 T2V 1.3B

Enderfga
authored and
SolitaryThinker
committed
[test] AnyFlow GPU smoke (pretrain + on-policy, 2 iters each)

Enderfga
authored and
SolitaryThinker
committed
[docs] AnyFlow algorithm + usage guide

Enderfga
authored and
SolitaryThinker
committed
[test] AnyFlow embedder tests: initialize ReplicatedLinear weights for CPU runs

Enderfga
authored and
SolitaryThinker
committed
[verify] AnyFlow↔FastVideo numerical parity script

Enderfga
authored and
SolitaryThinker
committed
[demo] AnyFlow 14B T2V inference demo (NFE=4 + NFE=50)

Enderfga
authored and
SolitaryThinker
committed
[misc] Drive AnyFlow helper script paths from env vars + ruff SIM108

Enderfga
authored and
SolitaryThinker
committed
[misc] Apply yapf/ruff pre-commit autofixes after rebase (main moved to line-length 120)
SolitaryThinker
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

[feat] Add AnyFlow any-step video distillation (pretrain + on-policy)#1371

[feat] Add AnyFlow any-step video distillation (pretrain + on-policy)#1371
mergify[bot] merged 16 commits into
hao-ai-lab:mainfrom
Enderfga:add-anyflow-pretrain-onpolicy

Commits on Jul 12, 2026

[feat] Add r_embedder config flags to Wan (defaults preserve bit-identity)

[feat] Extend WanTimeTextImageEmbedding with optional r_timestep

[feat] Thread optional r_timestep through WanTransformer3DModel.forward

[feat] Add FlowMapEulerDiscreteScheduler

[feat] AnyFlowPretrainMethod skeleton + (t, r) per-batch sampling

[test] AnyFlow pretrain: cover (t, r) sampling distribution + validation

[feat] AnyFlow pretrain: central-difference target + full single_train_step

[feat] AnyFlowMethod (on-policy DMD2 with multi-step Euler-flow rollout)

[feat] AnyFlow YAML configs (pretrain + on-policy) for Wan 2.1 T2V 1.3B

[test] AnyFlow GPU smoke (pretrain + on-policy, 2 iters each)

[docs] AnyFlow algorithm + usage guide

[test] AnyFlow embedder tests: initialize ReplicatedLinear weights for CPU runs

[verify] AnyFlow↔FastVideo numerical parity script

[demo] AnyFlow 14B T2V inference demo (NFE=4 + NFE=50)

[misc] Drive AnyFlow helper script paths from env vars + ruff SIM108

[misc] Apply yapf/ruff pre-commit autofixes after rebase (main moved to line-length 120)