Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Add a new "oumi fetch" command to pull configs via the CLI #1374

Open
taenin opened this issue Feb 4, 2025 · 0 comments
Open

[Feature] Add a new "oumi fetch" command to pull configs via the CLI #1374

taenin opened this issue Feb 4, 2025 · 0 comments
Labels

Comments

@taenin
Copy link
Collaborator

taenin commented Feb 4, 2025

Feature request

Create a new CLI command oumi fetch that will download configs from the oumi github repo to a local directory specified by an environment variable.

Any CLI actions using the oumi:// prefix will look for the config in that folder.

Motivation / references

This makes running Oumi much less disruptive. Users could even set the environment variable to a clone of the oumi repo to run the CLI anywhere. Consider the following:

OUMI_DIR=~/.oumi/configs


oumi infer -c oumi://smollm/inference/135m_infer.yaml \
  --generation.max_new_tokens 40 \
  --generation.temperature 0.7 \
  --interactive

Your contribution

This change requires:

  • Adding a new CLI entrypoint
  • Updating the config loading logic for each CLI entrypoint to check for the oumi:// prefix and resolve it accordingly.
@taenin taenin added the Feature label Feb 4, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

1 participant