blt/bytelatent/print_config.py
Pedro Rodriguez ece82cb960 Make it possible to specify multiple config files
Summary:

Test Plan:

Test that this iterpolates in the right order, config -> configs -> cli args

```
# All three sources
python -m bytelatent.print_config config=bytelatent/configs/debug.yaml configs=[internal/configs/s3_debug.yaml] eval=null

# What worked before
python -m bytelatent.print_config config=internal/configs/s3_debug.yaml eval=null
```
2025-02-12 19:25:06 +00:00

11 lines
191 B
Python

from bytelatent.args import TrainArgs, parse_args
def main():
train_args = parse_args(TrainArgs)
print(train_args.model_dump_json(indent=4))
if __name__ == "__main__":
main()