diff --git a/tutorials/asr/ASR_with_NeMo.ipynb b/tutorials/asr/ASR_with_NeMo.ipynb index 519456a012af..cf669f178845 100644 --- a/tutorials/asr/ASR_with_NeMo.ipynb +++ b/tutorials/asr/ASR_with_NeMo.ipynb @@ -1077,7 +1077,7 @@ "\n", "quartznet.export('qn.onnx')\n", "\n", - "ort_session = onnxruntime.InferenceSession('qn.onnx')\n", + "ort_session = onnxruntime.InferenceSession('qn.onnx', providers=['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider'])\n", "\n", "with tempfile.TemporaryDirectory() as tmpdir:\n", " with open(os.path.join(tmpdir, 'manifest.json'), 'w') as fp:\n", @@ -1172,4 +1172,4 @@ "outputs": [] } ] -} \ No newline at end of file +}