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

Browse Source

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).

...

This commit is contained in:

Nicolas Rabault

2026-06-22 15:43:52 +02:00

parent 73782447f2

commit 71c827f892

7 changed files with 179 additions and 2 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

tests/jobs/__init__.py