Export MeloTTS to ONNX (#1129)

This commit is contained in:
Fangjun Kuang
2024-07-15 10:47:19 +08:00
committed by GitHub
parent de04b3b9bf
commit 04c2319c2c
4 changed files with 573 additions and 0 deletions

41
scripts/melo-tts/run.sh Executable file
View File

@@ -0,0 +1,41 @@
#!/usr/bin/env bash
set -ex
function install() {
pip install torch==2.3.1+cpu torchaudio==2.3.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
pushd /tmp
git clone https://github.com/myshell-ai/MeloTTS
cd MeloTTS
pip install -r ./requirements.txt
pip install soundfile onnx onnxruntime
python3 -m unidic download
popd
}
install
export PYTHONPATH=/tmp/MeloTTS:$PYTHONPATH
echo "pwd: $PWD"
./export-onnx.py
ls -lh
head lexicon.txt
echo "---"
tail lexicon.txt
echo "---"
head tokens.txt
echo "---"
tail tokens.txt
./test.py
ls -lh