Support onnxruntime 1.18.0 (#906)

This commit is contained in:
Fangjun Kuang
2024-07-10 17:05:26 +08:00
committed by GitHub
parent 9e446b8501
commit dd0ff2ca06
107 changed files with 644 additions and 601 deletions

View File

@@ -9,10 +9,12 @@
set -ex
export PYTHONPATH=/tmp/wenet:$PYTHONPATH
function install_dependencies() {
pip install soundfile
pip install torch==2.1.0+cpu torchaudio==2.1.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
pip install k2==1.24.4.dev20231022+cpu.torch2.1.0 -f https://k2-fsa.github.io/k2/cpu.html
pip install torch==2.3.1+cpu torchaudio==2.3.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
pip install k2==1.24.4.dev20240606+cpu.torch2.3.1 -f https://k2-fsa.github.io/k2/cpu.html
pip install onnxruntime onnx kaldi-native-fbank pyyaml
@@ -31,7 +33,11 @@ function install_dependencies() {
cp -a ./wenet/wenet/ctl_model $wenet_dir
fi
rm -rf wenet
if [ ! -d $wenet_dir/finetune ]; then
cp -av ./wenet/wenet/finetune $wenet_dir/
fi
mv wenet /tmp
}
function aishell() {