Nicolas Rabault 71c827f892 feat(train): add JobConfig group, save_checkpoint_to_hub flag, Hub checkpoint helper ...

Introduce a JobConfig draccus group on TrainPipelineConfig (--job.target/image/
timeout/detach/tags) whose is_remote property gates remote dispatch, plus a
save_checkpoint_to_hub flag and validation. Add push_checkpoint_to_hub(), which
uploads a saved checkpoint directory to the model repo under checkpoints/<step>/
and creates the repo idempotently (private propagates from policy.private).

2026-06-22 16:24:05 +02:00

..

test_collate.py

Add extensive language support (#3467)

2026-05-19 14:46:11 +02:00

test_encoding_utils.py

chore(utils): move encoding utils and process to their respective modules (#2029)

2025-09-24 16:47:37 +02:00

test_io_utils.py

chore(docs): add missing license headers (#2140)

2025-10-08 14:27:52 +02:00

test_logging_utils.py

feat(training): bump accelerate + use reduction types for tracked metrics in a multi rank setup (#3773)

2026-06-11 19:07:28 +02:00

test_process.py

RL stack refactoring (#3075)

2026-05-12 15:49:54 +02:00

test_random_utils.py

chore(docs): add missing license headers (#2140)

2025-10-08 14:27:52 +02:00

test_replay_buffer.py

RL stack refactoring (#3075)

2026-05-12 15:49:54 +02:00

test_sample_weighting.py

Reward models refactor (#3142)

2026-04-28 17:56:24 +02:00

test_train_utils.py

feat(train): add JobConfig group, save_checkpoint_to_hub flag, Hub checkpoint helper

2026-06-22 16:24:05 +02:00

test_visualization_utils.py

feat(dependencies): minimal default tag install (#3362)

2026-04-12 20:03:04 +02:00