Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4
5base = AutoModelForCausalLM.from_pretrained(
6 "Qwen/Qwen2.5-1.5B-Instruct", torch_dtype=torch.float16, device_map="auto"
7)
8model = PeftModel.from_pretrained(base, "REPO_ID")
9tokenizer = AutoTokenizer.from_pretrained("REPO_ID")