William Lin 78d606a3a7 [feat]: make VSA tile cache configurable for training (#1444)

2026-06-28 02:18:31 -07:00

..

__init__.py

[Feat] Port LTX2 trainer (#1074)

2026-02-09 17:32:57 -08:00

activation_checkpoint.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

AGENTS.md

[misc] Rename MatrixGame to MatrixGame2 (#1357)

2026-05-17 22:57:03 -07:00

checkpointing_utils.py

[bugfix]Fixing Lora distillation training distributed checkpointing bug (#1192)

2026-04-06 02:20:26 +00:00

cosmos2_5_training_pipeline.py

[feat] Add Cosmos 2.5 T2W training pipeline (LoRA + full fine-tune) (#1227)

2026-05-11 19:15:00 -07:00

distillation_pipeline.py

[bugfix] EMA shadow on resume and EMA under MoE path (#1441)

2026-06-09 00:55:43 +00:00

ltx2_training_pipeline.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

matrixgame2_ar_diffusion_pipeline.py

[model] Flux2 Klein Port (#1349)

2026-06-09 14:55:55 -07:00

matrixgame2_ode_causal_pipeline.py

[model] Flux2 Klein Port (#1349)

2026-06-09 14:55:55 -07:00

matrixgame2_self_forcing_distillation_pipeline.py

[model] Flux2 Klein Port (#1349)

2026-06-09 14:55:55 -07:00

matrixgame2_training_pipeline.py

[model] Flux2 Klein Port (#1349)

2026-06-09 14:55:55 -07:00

ode_causal_pipeline.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

self_forcing_distillation_pipeline.py

[bugfix] EMA shadow on resume and EMA under MoE path (#1441)

2026-06-09 00:55:43 +00:00

trackers.py

[feat] Add Wan RL DiffusionNFT training (#1450)

2026-06-11 21:18:59 -07:00

training_pipeline.py

[feat]: make VSA tile cache configurable for training (#1444)

2026-06-28 02:18:31 -07:00

training_utils.py

[feat] Add Wan RL DiffusionNFT training (#1450)

2026-06-11 21:18:59 -07:00

wan_distillation_pipeline.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

wan_i2v_distillation_pipeline.py

[model] Flux2 Klein Port (#1349)

2026-06-09 14:55:55 -07:00

wan_i2v_training_pipeline.py

[model] Flux2 Klein Port (#1349)

2026-06-09 14:55:55 -07:00

wan_self_forcing_distillation_pipeline.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

wan_training_pipeline.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00