cassuto
/

Llama-3.1-ANLI-R1-R2-R3-8B-Instruct

Sentence Similarity

Textual Entailment

Model card Files Files and versions

cassuto commited on Nov 15, 2024

Commit

7d6bf6f

·

verified ·

1 Parent(s): 811482d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ import torch
 base_model_name = 'meta-llama/Llama-3.1-8B-Instruct'
 tokenizer = AutoTokenizer.from_pretrained(base_model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name,
                 pad_token_id=tokenizer.eos_token_id,
                 device_map='auto')

 base_model_name = 'meta-llama/Llama-3.1-8B-Instruct'
 tokenizer = AutoTokenizer.from_pretrained(base_model_name)
+model = AutoModelForCausalLM.from_pretrained(base_model_name,
                 pad_token_id=tokenizer.eos_token_id,
                 device_map='auto')