How to use thirtyninetythree/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, "thirtyninetythree/Sheared-LLaMA-1.3B-summarization")
The community tab is the place to discuss and collaborate with the HF community!