How to use Vishal24/trained-llama2 with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("Vishal24/llama2-7B-base") model = PeftModel.from_pretrained(base_model, "Vishal24/trained-llama2")
What is a pickle import?
