Skip to content

Commit

Permalink
use cmd line
Browse files Browse the repository at this point in the history
  • Loading branch information
v-chen_data committed Jul 3, 2024
1 parent 3fbe51f commit 868e55b
Showing 1 changed file with 2 additions and 11 deletions.
13 changes: 2 additions & 11 deletions llmfoundry/train/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -565,19 +565,10 @@ def main(cfg: DictConfig) -> Trainer:


@app.command()
def train(
yaml_path: str = typer.Argument(
...,
help='Path to the YAML configuration file',
),
args_list: List[str] = typer.Option(
None,
help='Additional command line arguments',
hidden=True,
),
):
def train():
"""Run the training with the given configuration and optional overrides from
command line."""
yaml_path, args_list = sys.argv[1], sys.argv[2:]
# Disable resolving environment variables through omegaconf.
om.clear_resolver('oc.env')

Expand Down

0 comments on commit 868e55b

Please sign in to comment.