We use the architecture and tokenizer of TinyLlama v1.1.
Please use transformers>=4.35.
Models
The main branch of the repository contains the best-performing model that was evaluated in the paper. Other checkpoints produced during training are also hosted in this repository under different branch names (also called "revisions" in HuggingFace), with each branch name referring to the number of training steps.