Add AnyFlow algorithm (any-step video diffusion via flow maps) - #25

Merged

juliusberner merged 19 commits into

NVlabs:mainfrom

Enderfga:feature/anyflow-algorithm

Aug 21, 2026

Commits on Aug 19, 2026

Add AnyFlow algorithm

Enderfga
authored and
juliusberner
committed
b2d7cdd
Wan: add gated r-embedder fusion + AnyFlow weight remap

Enderfga
authored and
juliusberner
committed
aec727d
anyflow: multi-step rollout-with-gradient on-policy student generation

Enderfga
authored and
juliusberner
committed
bcd78d6
Wan: extract _fuse_r_embedding helper; ship AnyFlow on-policy config

Enderfga
authored and
juliusberner
committed
5437b4b
anyflow: reuse MeanFlow for pretrain and stock DMD2 for on-policy

Enderfga
authored and
juliusberner
committed
994aad1
anyflow: align training math with the reference implementation

Enderfga
authored and
juliusberner
committed
667c47d
anyflow: keep DMD-to-flow-map gradient ratio at the reference's 1:1

Enderfga
authored and
juliusberner
committed
6b097d7
anyflow: address second-round review

Enderfga
authored and
juliusberner
committed
a3fd37e
anyflow: validate guidance-fusion inputs

Enderfga
authored and
juliusberner
committed
01c8ff2
anyflow: reduce rebalance scalars; warn on degenerate buckets

Enderfga
authored and
juliusberner
committed
b6c153e
anyflow: drop the rank-local guard around the rebalance collective

Enderfga
authored and
juliusberner
committed
3dae9d9
anyflow: fix stale docs

Enderfga
authored and
juliusberner
committed
01be71f
anyflow: reference-exact [0,1] timestep range, on-policy noise start, doc trims
juliusberner
committed
e3d2833
anyflow: minor edits
juliusberner
committed
5d5916d
Finalize integration and configs
juliusberner
committed
23c6ca1
Address comments
juliusberner
committed
68a83a4
Lint
juliusberner
committed
a6b2497

Commits on Aug 20, 2026

anyflow: restore the guidance-fusion neg_condition guard

Enderfga
authored and
juliusberner
committed
2a1a495
anyflow: split co-train sampling cfg, factor time_shift and train_mode
juliusberner
committed
63ec082

Navigation Menu

Uh oh!

Add AnyFlow algorithm (any-step video diffusion via flow maps) - #25

Add AnyFlow algorithm (any-step video diffusion via flow maps)#25
juliusberner merged 19 commits into
NVlabs:mainfrom
Enderfga:feature/anyflow-algorithm

Commits on Aug 19, 2026

Add AnyFlow algorithm

Wan: add gated r-embedder fusion + AnyFlow weight remap

anyflow: multi-step rollout-with-gradient on-policy student generation

Wan: extract _fuse_r_embedding helper; ship AnyFlow on-policy config

anyflow: reuse MeanFlow for pretrain and stock DMD2 for on-policy

anyflow: align training math with the reference implementation

anyflow: keep DMD-to-flow-map gradient ratio at the reference's 1:1

anyflow: address second-round review

anyflow: validate guidance-fusion inputs

anyflow: reduce rebalance scalars; warn on degenerate buckets

anyflow: drop the rank-local guard around the rebalance collective

anyflow: fix stale docs

anyflow: reference-exact [0,1] timestep range, on-policy noise start, doc trims

anyflow: minor edits

Finalize integration and configs

Address comments

Lint

Commits on Aug 20, 2026

anyflow: restore the guidance-fusion neg_condition guard

anyflow: split co-train sampling cfg, factor time_shift and train_mode