"""Compile the bundled ROM into a standalone, stock Transformers checkpoint."""
import argparse
from pathlib import Path
from beamboot.causal_lm import CausalMachine
from beamboot.os import firmware


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('output', type=Path, help='output directory for config and safetensors')
    args = parser.parse_args()
    model = CausalMachine(firmware()[0])
    model.save_pretrained(args.output)
    print(f'Saved constructed Beam Boot checkpoint to {args.output}')


if __name__ == '__main__': main()