from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer

base = "meta-llama/Meta-Llama-3.1-8B-Instruct"
adapter = "REPO_ID"

tokenizer = AutoTokenizer.from_pretrained(base, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
    base,
    device_map="auto",
    torch_dtype="auto",
    trust_remote_code=True,
)
model = PeftModel.from_pretrained(model, adapter)

Roseshmay/led-qa-llama-3.1-8b-lora

LED QA LoRA — Llama-3.1-8B-Instruct

Base model

Training

Load

Roseshmay/
led-qa-llama-3.1-8b-lora