Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4base_model = AutoModelForCausalLM.from_pretrained(
5 "meta-llama/Llama-3.1-8B-Instruct", torch_dtype="auto", device_map="auto",
6)
7model = PeftModel.from_pretrained(base_model, "<path-to-adapter>")
8tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.1-8B-Instruct")| Parameter | Value |
|---|---|
| Base model | meta-llama/Llama-3.1-8B-Instruct |
| LoRA rank | 4 |
| LoRA alpha | 8 |
| Dropout | 0.1 |
| Target modules | down_proj, gate_proj, k_proj, up_proj, v_proj, o_proj, q_proj |
| PEFT version | 0.18.0 |