Views
No views yet
transformers>=4.36.1
peft>=0.6.21from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4base_model = AutoModelForCausalLM.from_pretrained('meta-llama/Llama-2-7b-hf')
5base_model.resize_token_embeddings(260164)
6tokenizer = AutoTokenizer.from_pretrained('MaLA-LM/mala-500')
7model = PeftModel.from_pretrained(base_model, 'MaLA-LM/mala-500')@misc{lin2024mala500,
title={MaLA-500: Massive Language Adaptation of Large Language Models},
author={Peiqin Lin and Shaoxiong Ji and Jörg Tiedemann and André F. T. Martins and Hinrich Schütze},
year={2024},
eprint={2401.13303},
archivePrefix={arXiv},
primaryClass={cs.CL}
}