SetFit/rte
Viewer โข Updated โข 5.77k โข 2.53k โข 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