Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4base = AutoModelForCausalLM.from_pretrained(
5 "google/gemma-3-1b-it", torch_dtype=torch.bfloat16, device_map="auto"
6)
7model = PeftModel.from_pretrained(base, "REPO_ID")
8tokenizer = AutoTokenizer.from_pretrained("REPO_ID")