Hugging Face's logo

Spaces:
YUNSUN7
/
Applioceec
Build error

App Files Community
YUNSUN7 commited on
Commit
d65e6aa
·
verified ·
1 Parent(s): dc8f793

Create Cuda

Browse files
Files changed (1) hide show
  1. Cuda +7 -0
Cuda ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ from optimum.onnxruntime import ORTModelForSequenceClassification
2
+
3
+ ort_model = ORTModelForSequenceClassification.from_pretrained(
4
+ "distilbert/distilbert-base-uncased-finetuned-sst-2-english",
5
+ export=True,
6
+ provider="CUDAExecutionProvider",
7
+ )