blt/bytelatent/print_config.py
Pedro Rodriguez bec0164820 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-14 21:03:57 +00:00

12 lines
263 B
Python

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