Views
No views yet
meta-llama/Meta-Llama-3.1-8B-Instructmerge_and_unload()messages1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3repo_id = "Sara121/Llama-3.1-8B-ABS-Regulatory"
4
5tokenizer = AutoTokenizer.from_pretrained(repo_id)
6model = AutoModelForCausalLM.from_pretrained(
7 repo_id,
8 torch_dtype="auto",
9 device_map="auto",
10)
11
12model.eval()