fix onnx export tests

k2-fsa · Sep 7, 2024 · a1033e3 · a1033e3
1 parent da6e0a8
commit a1033e3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/.github/scripts/test-onnx-export.sh b/.github/scripts/test-onnx-export.sh
@@ -25,6 +25,7 @@ popd
 
 log "Export via torch.jit.script()"
 ./zipformer/export.py \
+  --use-averaged-model 0 \
   --exp-dir $repo/exp \
   --tokens $repo/data/lang_bpe_500/tokens.txt \
   --epoch 99 \

diff --git a/.github/workflows/test-onnx-export.yml b/.github/workflows/test-onnx-export.yml
@@ -4,6 +4,7 @@ on:
   push:
     branches:
       - master
+      - fix-ci
   pull_request:
     types: [labeled]
 
@@ -16,6 +17,8 @@ on:
     # nightly build at 15:50 UTC time every day
     - cron: "50 15 * * *"
 
+  workflow_dispatch:
+
 concurrency:
   group: test_onnx_export-${{ github.ref }}
   cancel-in-progress: true