How to use karanjaxyz/Sheared-LLaMA-1.3B-summarization with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("princeton-nlp/Sheared-LLaMA-1.3B") model = PeftModel.from_pretrained(base_model, "karanjaxyz/Sheared-LLaMA-1.3B-summarization")
