MaLA-LM
/

mala-500-10b-v1

Text Generation

Inference Endpoints

Model card Files Files and versions Community

jisx commited on Jan 25

Commit

9baff34

•

1 Parent(s): f6266f1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -21,7 +21,7 @@ Use the code below to get started with the model.
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
-base_model = AutoModelForCausalLM.from_pretrained('meta-llama/Llama-2-7b')
 base_model.resize_token_embeddings(260164)
 tokenizer = AutoTokenizer.from_pretrained('MaLA-LM/mala-500')
 model = PeftModel.from_pretrained(base_model, 'MaLA-LM/mala-500')

 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
+base_model = AutoModelForCausalLM.from_pretrained('meta-llama/Llama-2-7b-hf')
 base_model.resize_token_embeddings(260164)
 tokenizer = AutoTokenizer.from_pretrained('MaLA-LM/mala-500')
 model = PeftModel.from_pretrained(base_model, 'MaLA-LM/mala-500')