Views
No views yet
peft library with the base model:1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from peft import PeftModel
4
5model_id = "LGAI-EXAONE/EXAONE-Deep-7.8B"
6adapter_id = "umyunsang/civil-complaint-exaone-lora"
7
8tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
9model = AutoModelForCausalLM.from_pretrained(
10 model_id,
11 torch_dtype=torch.bfloat16,
12 device_map="auto",
13 trust_remote_code=True
14)
15model = PeftModel.from_pretrained(model, adapter_id)