SetFit/rte
Viewer • Updated • 5.77k • 2.08k • 2
How to use yuuhan/roberta-base-rte-lora with PEFT:
from peft import PeftModel
from transformers import AutoModelForSequenceClassification
base_model = AutoModelForSequenceClassification.from_pretrained("/home/sunyuhan/syh/sunyuhan/zju/roberta-base")
model = PeftModel.from_pretrained(base_model, "yuuhan/roberta-base-rte-lora")This modelcard aims to be a base template for new models. It has been generated using this raw template.
Accuracy: 0.7328519855595668 on RTE