From 3553e5aec51411483de6d49da573efb07cc38d14 Mon Sep 17 00:00:00 2001 From: fayejf Date: Thu, 12 Oct 2023 00:43:34 -0700 Subject: [PATCH] fix onnx Signed-off-by: fayejf --- tutorials/nlp/Text_Classification_Sentiment_Analysis.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorials/nlp/Text_Classification_Sentiment_Analysis.ipynb b/tutorials/nlp/Text_Classification_Sentiment_Analysis.ipynb index a8ebf05dc607..2570eac5100f 100644 --- a/tutorials/nlp/Text_Classification_Sentiment_Analysis.ipynb +++ b/tutorials/nlp/Text_Classification_Sentiment_Analysis.ipynb @@ -774,7 +774,7 @@ "model.eval()\n", "infer_datalayer = create_infer_dataloader(model, queries)\n", "\n", - "ort_session = onnxruntime.InferenceSession(config.export_to)\n", + "ort_session = onnxruntime.InferenceSession(config.export_to, providers=['CPUExecutionProvider'])\n", "\n", "for batch in infer_datalayer:\n", " input_ids, input_type_ids, input_mask, subtokens_mask = batch\n",