From 1f32ca6c9b18c55355d459c73426b7a17e586e09 Mon Sep 17 00:00:00 2001 From: Edward Chen <18449977+edgchen1@users.noreply.github.com> Date: Mon, 14 Oct 2024 14:31:06 -0700 Subject: [PATCH] Update mobile/examples/speech_recognition/model dependencies. (#480) Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> --- .../examples/speech_recognition/model/requirements.txt | 9 ++++----- mobile/examples/speech_recognition/model/wav2vec2_gen.py | 2 +- 2 files changed, 5 insertions(+), 6 deletions(-) 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)