a2rl.experimental.lightgpt.LightGPTBuilder.save_model

LightGPTBuilder.save_model()

Save trained pytorch model, training config, and associated tokenizer.

Tokenizer and training config will be saved as “tokenizer.pt” and “config.yaml” respectively.