alexzms b867ef7e7c [SP Sharding] Fix SP loss sharding on token axis (thw) with padding; add distributed correctness tests (#1045) ...

Fixes the sequence parallel sharding on t, now SP shards on t*h*w

2026-01-27 22:53:04 -08:00

..

__init__.py

[Feature] Add DMD T2V training pipeline (#651)

2025-07-27 03:35:51 -04:00

activation_checkpoint.py

[Feature] Remove V1 folder (#642)

2025-07-24 22:43:12 -07:00

checkpointing_utils.py

[Feature] Remove V1 folder (#642)

2025-07-24 22:43:12 -07:00

distillation_pipeline.py

[bugfix] [dmd2] allow dmd2 simulate_student_forward to use text-only dataset (#951)

2025-12-23 00:41:31 -06:00

matrixgame_training_pipeline.py

[feature] Add Matrix Game 2.0 training (#1017)

2026-01-26 19:21:43 -08:00

ode_causal_pipeline.py

[feat] unified trainer logging (#841)

2025-10-16 23:16:16 -07:00

self_forcing_distillation_pipeline.py

[CI] Fix CI tests (#935)

2025-12-16 04:59:43 -08:00

trackers.py

[feat] unified trainer logging (#841)

2025-10-16 23:16:16 -07:00

training_pipeline.py

[SP Sharding] Fix SP loss sharding on token axis (thw) with padding; add distributed correctness tests (#1045)

2026-01-27 22:53:04 -08:00

training_utils.py

[SP Sharding] Fix SP loss sharding on token axis (thw) with padding; add distributed correctness tests (#1045)

2026-01-27 22:53:04 -08:00

wan_distillation_pipeline.py

[self-forcing][8/n] Self-Forcing For Wan2.2-A14B + torch.compile training and distillation support (#818)

2025-10-02 15:01:45 -07:00

wan_i2v_distillation_pipeline.py

[self-forcing][8/n] Self-Forcing For Wan2.2-A14B + torch.compile training and distillation support (#818)

2025-10-02 15:01:45 -07:00

wan_i2v_training_pipeline.py

[Bugfix] Add i2v vae loading (#686)

2025-08-01 23:15:37 -04:00

wan_self_forcing_distillation_pipeline.py

[self-forcing][8/n] Self-Forcing For Wan2.2-A14B + torch.compile training and distillation support (#818)

2025-10-02 15:01:45 -07:00

wan_training_pipeline.py

[Fix] training pipeline pin_cpu_memory issue (#692)

2025-08-07 02:31:20 -07:00