Couple of tweaks for asr2.sh for the HF hub upload

espnet · Jul 23, 2023 · a00b287 · a00b287
1 parent 890ba71
commit a00b287
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/egs2/TEMPLATE/asr2/asr2.sh b/egs2/TEMPLATE/asr2/asr2.sh
@@ -1661,11 +1661,14 @@ if [ ${stage} -le 18 ] && [ ${stop_stage} -ge 18 ] && ! [[ " ${skip_stages} " =~
     espnet_task=ASR
     # shellcheck disable=SC2034
     task_exp=${asr_exp}
+    # shellcheck disable=SC2034
+    lang=${tgt_lang}
     eval "echo \"$(cat scripts/utils/TEMPLATE_HF_Readme.md)\"" > "${dir_repo}"/README.md
 
     this_folder=${PWD}
     cd ${dir_repo}
     if [ -n "$(git status --porcelain)" ]; then
+        git lfs track *.mdl
         git add .
         git commit -m "Update model"
     fi