Skip to content

[CLI] Extend training support to all trainers #2101

@lewtun

Description

@lewtun

Feature request

The CLI currently supports training models with SFT/DPO/KTO:

SUPPORTED_COMMANDS = ["sft", "dpo", "chat", "kto"]

It would be good to extend this support to all trainers so that we have a consistent API and also learn which parts of our scripts need refactoring to support this usage.

This could be tackled in separate PRs to keep things lightweight, and I'll track here the trainers in terms of priority to add (based on Hub usage):

Motivation

It is somewhat annoying that one cannot train a model through the CLI as this is helpful for fast debugging / iterations.

Your contribution

Happy to open PRs, but this could be a good first issue for new contributors!

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions