TinyLlama-1.1B fine-tuned on MetaMathQA using a from-scratch LoRA implementation.
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("schwp/schwp/TinyMathLlama-1.1B")
The training and evaluation scripts are available on this
github repository.
The whole LoRA implementation used for the fine-tuning is also on the repository.