Views
No views yet
1from transformers import AutoModelForCausalLM
2from peft import PeftModel
3
4base = AutoModelForCausalLM.from_pretrained(
5 "LGAI-EXAONE/EXAONE-4.0-1.2B",
6 torch_dtype="bfloat16",
7 device_map="auto",
8)
9
10model = PeftModel.from_pretrained(
11 base,
12 "USERNAME/exaone-4.0-1.2b-vulnreport-lora",
13 is_trainable=False
14)