Will Lin 141a1140f6 refactor sampling pipeline

2026-01-20 15:40:53 -08:00

..

rl

refactor sampling pipeline

2026-01-20 15:40:53 -08:00

__init__.py

Phase 1 minor fixes

2026-01-20 14:38:17 -08: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

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

fix trajectory collection & reward computation

2026-01-20 14:38:20 -08:00

training_utils.py

[feat] Support sequence packing and shard after pachification for USP (#894)

2025-12-19 16:19:46 -06: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_rl_training_pipeline.py

refactor sampling pipeline

2026-01-20 15:40:53 -08: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