optimum-internal-testing/optimum-neuron-testing-0.4.4.dev2-2.26.1-trn1-c40fc1a413-speculation-draft
Updated
•
12
None defined yet.
from optimum.onnxruntime import ORTModelForSequenceClassification
# Load the model from the hub and export it to the ONNX format
model_id = "distilbert-base-uncased-finetuned-sst-2-english"
model = ORTModelForSequenceClassification.from_pretrained(model_id, export=True)