Skip to content

Commit

Permalink
test more models
Browse files Browse the repository at this point in the history
  • Loading branch information
csukuangfj committed Dec 20, 2023
1 parent f37a75e commit 9f73b87
Showing 1 changed file with 10 additions and 8 deletions.
18 changes: 10 additions & 8 deletions .github/workflows/export-whisper-to-onnx.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ jobs:
- name: Install dependencies
shell: bash
run: |
python3 -m pip install torch==1.13.0 -f https://download.pytorch.org/whl/cpu/torch_stable.html
python3 -m pip install torch==1.13.0 torchaudio==0.13.0 -f https://download.pytorch.org/whl/cpu/torch_stable.html
python3 -m pip install openai-whisper==20230314 onnxruntime onnx
- name: export ${{ matrix.model }}
Expand Down Expand Up @@ -108,15 +108,17 @@ jobs:
repo_token: ${{ secrets.UPLOAD_GH_SHERPA_ONNX_TOKEN }}
tag: asr-models

- name: Test tiny.en
if: matrix.model == 'tiny.en'
- name: Test ${{ matrix.model }}
shell: bash
run: |
src=sherpa-onnx-whisper-tiny.en
python3 srcipts/whisper/test.py \
--encoder $src/tiny.en-encoder.int8.onnx \
--decoder $src/tiny.en-decoder.int8.onnx \
--tokens $src/tiny.en-tokens.txt \
python3 -m pip install kaldi-native-fbank
git checkout .
model=${{ matrix.model }}
src=sherpa-onnx-whisper-$model
python3 scripts/whisper/test.py \
--encoder $src/$model-encoder.int8.onnx \
--decoder $src/$model-decoder.int8.onnx \
--tokens $src/$model-tokens.txt \
$src/test_wavs/0.wav
- name: Publish ${{ matrix.model }} to huggingface
Expand Down

0 comments on commit 9f73b87

Please sign in to comment.