Salesforce/wikitext
Viewer • Updated • 3.71M • 1.35M • 694
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
model_name = "ziyingchen1106/Llama-3.2-3B-Instruct-fp16-lora-wikitext"
model = AutoModelForCausalLM.from_pretrained(
model_name,
torch_dtype=torch.float16,
device_map="cuda:0"
)
tokenizer = AutoTokenizer.from_pretrained(model_name)
Base model
meta-llama/Llama-3.2-3B-Instruct