Spaces:
Build error
Build error
File size: 246 Bytes
d65e6aa |
1 2 3 4 5 6 7 |
from optimum.onnxruntime import ORTModelForSequenceClassification
ort_model = ORTModelForSequenceClassification.from_pretrained(
"distilbert/distilbert-base-uncased-finetuned-sst-2-english",
export=True,
provider="CUDAExecutionProvider",
) |