From 92eb8bc22f8c5021b21df6c5e842e551e6c8d566 Mon Sep 17 00:00:00 2001
From: jinzr <60612200+JinZr@users.noreply.github.com>
Date: Thu, 7 Sep 2023 19:45:02 +0800
Subject: [PATCH] Update run-multi-zh_hans-zipformer.sh

---
 .github/scripts/run-multi-zh_hans-zipformer.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/scripts/run-multi-zh_hans-zipformer.sh b/.github/scripts/run-multi-zh_hans-zipformer.sh
index 80cf4b160..6f40b98b6 100755
--- a/.github/scripts/run-multi-zh_hans-zipformer.sh
+++ b/.github/scripts/run-multi-zh_hans-zipformer.sh
@@ -30,7 +30,7 @@ log "Test exporting to ONNX format"
 
 ./zipformer/export.py \
   --exp-dir $repo/exp \
-  --lang-dir $repo/data/lang_bpe_2000 \
+  --tokens $repo/data/lang_bpe_2000/tokens.txt \
   --epoch 99 \
   --avg 1 \
   --onnx 1
@@ -39,14 +39,14 @@ log "Export to torchscript model"
 
 ./zipformer/export.py \
   --exp-dir $repo/exp \
-  --lang-dir $repo/data/lang_bpe_2000 \
+  --tokens $repo/data/lang_bpe_2000/tokens.txt \
   --epoch 99 \
   --avg 1 \
   --jit 1
 
 ./zipformer/export.py \
   --exp-dir $repo/exp \
-  --lang-dir $repo/data/lang_bpe_2000 \
+  --tokens $repo/data/lang_bpe_2000/tokens.txt \
   --epoch 99 \
   --avg 1 \
   --jit-trace 1