How to use AmberJin4526/distilbert-base-uncased-lora-text-classification with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased") model = PeftModel.from_pretrained(base_model, "AmberJin4526/distilbert-base-uncased-lora-text-classification")
How to use AmberJin4526/distilbert-base-uncased-lora-text-classification with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("AmberJin4526/distilbert-base-uncased-lora-text-classification", device_map="auto")
