diff --git a/mobile/examples/speech_recognition/model/requirements.txt b/mobile/examples/speech_recognition/model/requirements.txt index bc341531..beac372c 100644 --- a/mobile/examples/speech_recognition/model/requirements.txt +++ b/mobile/examples/speech_recognition/model/requirements.txt @@ -1,6 +1,5 @@ -onnx==1.13.0 -onnxruntime==1.13.1 -protobuf==3.20.2 -torch==1.13.1 -torchaudio==0.13.1 +onnx==1.16.1 +onnxruntime==1.19.2 +torch==2.2.2 +torchaudio==2.2.2 transformers==4.38.0 diff --git a/mobile/examples/speech_recognition/model/wav2vec2_gen.py b/mobile/examples/speech_recognition/model/wav2vec2_gen.py index c92704c3..d8f14cee 100644 --- a/mobile/examples/speech_recognition/model/wav2vec2_gen.py +++ b/mobile/examples/speech_recognition/model/wav2vec2_gen.py @@ -19,4 +19,4 @@ input_names=["input"], output_names=["output"], dynamic_axes={"input": [1], "output": [1]}, - opset_version=13) + opset_version=14)