-
Notifications
You must be signed in to change notification settings - Fork 570
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Export FireRedASR to sherpa-onnx. (#1865)
- Loading branch information
1 parent
60beff1
commit 2337169
Showing
2 changed files
with
120 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,119 @@ | ||
name: export-fire-red-asr-to-onnx | ||
|
||
on: | ||
push: | ||
branches: | ||
- export-fire-red-asr | ||
|
||
workflow_dispatch: | ||
|
||
concurrency: | ||
group: export-fire-red-asr-to-onnx-${{ github.ref }} | ||
cancel-in-progress: true | ||
|
||
jobs: | ||
export-fire-red-asr-to-onnx: | ||
if: github.repository_owner == 'k2-fsa' || github.repository_owner == 'csukuangfj' | ||
name: export FireRedAsr ${{ matrix.version }} | ||
runs-on: ${{ matrix.os }} | ||
strategy: | ||
fail-fast: false | ||
matrix: | ||
os: [ubuntu-latest] | ||
python-version: ["3.10"] | ||
|
||
steps: | ||
- uses: actions/checkout@v4 | ||
|
||
- name: Setup Python ${{ matrix.python-version }} | ||
uses: actions/setup-python@v5 | ||
with: | ||
python-version: ${{ matrix.python-version }} | ||
|
||
- name: Install Python dependencies | ||
shell: bash | ||
run: | | ||
pip install "numpy<=1.26.4" onnx==1.16.0 onnxruntime==1.17.1 | ||
- name: Download exported ONNX model from ModelScope | ||
env: | ||
MS_TOKEN: ${{ secrets.MODEL_SCOPE_GIT_TOKEN }} | ||
shell: bash | ||
run: | | ||
git clone https://oauth2:${MS_TOKEN}@www.modelscope.cn/csukuangfj/sherpa-onnx-fire-red-asr-large-zh_en-2025-02-16.git ms | ||
ls -lh ms | ||
- name: Collect results | ||
shell: bash | ||
run: | | ||
src=ms | ||
d=sherpa-onnx-fire-red-asr-large-zh_en-2025-02-16 | ||
mkdir $d | ||
mv -v $src/*.onnx $d | ||
cp -v $src/README.md $d | ||
cp -v $src/tokens.txt $d | ||
cp -av $src/test_wavs $d | ||
ls -lh $d/ | ||
tar cjfv $d.tar.bz2 $d | ||
ls -lh $d.tar.bz2 | ||
rm -rf ms | ||
- name: Publish to huggingface ${{ matrix.version }} | ||
env: | ||
HF_TOKEN: ${{ secrets.HF_TOKEN }} | ||
uses: nick-fields/retry@v3 | ||
with: | ||
max_attempts: 20 | ||
timeout_seconds: 200 | ||
shell: bash | ||
command: | | ||
git config --global user.email "[email protected]" | ||
git config --global user.name "Fangjun Kuang" | ||
rm -rf huggingface | ||
export GIT_LFS_SKIP_SMUDGE=1 | ||
export GIT_CLONE_PROTECTION_ACTIVE=false | ||
src=sherpa-onnx-fire-red-asr-large-zh_en-2025-02-16 | ||
git clone https://csukuangfj:[email protected]/csukuangfj/$src huggingface | ||
cd huggingface | ||
rm -rf ./* | ||
git fetch | ||
git pull | ||
cp -av ../$src/* ./ | ||
git lfs track "*.onnx" | ||
git add . | ||
ls -lh | ||
git status | ||
git commit -m "add models" | ||
git push https://csukuangfj:[email protected]/csukuangfj/$src main || true | ||
- name: Release | ||
if: github.repository_owner == 'csukuangfj' | ||
uses: svenstaro/upload-release-action@v2 | ||
with: | ||
file_glob: true | ||
file: ./*.tar.bz2 | ||
overwrite: true | ||
repo_name: k2-fsa/sherpa-onnx | ||
repo_token: ${{ secrets.UPLOAD_GH_SHERPA_ONNX_TOKEN }} | ||
tag: asr-models | ||
|
||
- name: Release | ||
if: github.repository_owner == 'k2-fsa' | ||
uses: svenstaro/upload-release-action@v2 | ||
with: | ||
file_glob: true | ||
file: ./*.tar.bz2 | ||
overwrite: true | ||
tag: asr-models |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters