How to use viv6267/Test_sagemaker with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("NousResearch/Llama-2-7b-hf") model = PeftModel.from_pretrained(base_model, "viv6267/Test_sagemaker")
