From bb8cf2d9eb107442e597b052fbec4cd1c8196501 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 2 Jan 2025 11:52:02 +0800 Subject: [PATCH] fix use_eos_bos --- .github/scripts/ljspeech/TTS/run-matcha.sh | 2 +- .github/workflows/ljspeech.yml | 4 ++-- egs/ljspeech/TTS/matcha/export_onnx.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/scripts/ljspeech/TTS/run-matcha.sh b/.github/scripts/ljspeech/TTS/run-matcha.sh index f56b101b8..bfb37fb6d 100755 --- a/.github/scripts/ljspeech/TTS/run-matcha.sh +++ b/.github/scripts/ljspeech/TTS/run-matcha.sh @@ -77,7 +77,7 @@ function export_onnx() { popd pushd data/fbank - rm -v *.json + rm -fv *.json curl -SL -O https://huggingface.co/csukuangfj/icefall-tts-ljspeech-matcha-en-2024-10-28/resolve/main/data/cmvn.json popd diff --git a/.github/workflows/ljspeech.yml b/.github/workflows/ljspeech.yml index 56ba68a57..31a65cd94 100644 --- a/.github/workflows/ljspeech.yml +++ b/.github/workflows/ljspeech.yml @@ -130,7 +130,7 @@ jobs: path: ./model-steps-6.onnx - name: Upload models to huggingface - if: matrix.python-version == '3.9' && matrix.torch-version == '2.3.0' && github.event_name == 'push' + if: matrix.python-version == '3.9' && matrix.torch-version == '2.3.0' shell: bash env: HF_TOKEN: ${{ secrets.HF_TOKEN }} @@ -155,7 +155,7 @@ jobs: popd - name: Release exported onnx models - if: matrix.python-version == '3.9' && matrix.torch-version == '2.3.0' && github.event_name == 'push' + if: matrix.python-version == '3.9' && matrix.torch-version == '2.3.0' uses: svenstaro/upload-release-action@v2 with: file_glob: true diff --git a/egs/ljspeech/TTS/matcha/export_onnx.py b/egs/ljspeech/TTS/matcha/export_onnx.py index 3473c08f5..3c653fbf1 100755 --- a/egs/ljspeech/TTS/matcha/export_onnx.py +++ b/egs/ljspeech/TTS/matcha/export_onnx.py @@ -183,7 +183,7 @@ def main(): "pad_id": tokenizer.pad_id, "model_author": "icefall", "maintainer": "k2-fsa", - "use_eos_bos": 0, + "use_eos_bos": 1, "dataset": "LJ Speech", "dataset_url": "https://keithito.com/LJ-Speech-Dataset/", "num_ode_steps": num_steps,