Support onnxruntime 1.18.0 (#906)
This commit is contained in:
@@ -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() {
|
||||
|
||||
Reference in New Issue
Block a user