Skip to content

Commit dde688a

Browse files
author
Nithin Rao Koluguri
committed
add latest parakeet-v2
Signed-off-by: Nithin Rao Koluguri <nithinraok>
1 parent 59dd8be commit dde688a

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

nemo_asr/run_fast_conformer_rnnt.sh

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,11 @@
22

33
export PYTHONPATH="..":$PYTHONPATH
44

5-
#considering FC-L, FC-XL, FC-XXL, C-L and C-S RNNT models
6-
MODEL_IDs=("nvidia/parakeet-tdt-1.1b" "nvidia/parakeet-rnnt-1.1b" "nvidia/parakeet-rnnt-0.6b" "nvidia/stt_en_fastconformer_transducer_large" "nvidia/stt_en_conformer_transducer_large" "stt_en_conformer_transducer_small")
7-
BATCH_SIZE=64
5+
#considering latest model
6+
MODEL_IDs=("nvidia/parakeet-tdt-0.6b-v2")
7+
# For previous parakeet models: FC-L, FC-XL, FC-XXL, C-L and C-S RNNT models
8+
# ("nvidia/parakeet-tdt-1.1b" "nvidia/parakeet-rnnt-1.1b" "nvidia/parakeet-rnnt-0.6b" "nvidia/stt_en_fastconformer_transducer_large" "nvidia/stt_en_conformer_transducer_large" "stt_en_conformer_transducer_small")
9+
BATCH_SIZE=128
810
DEVICE_ID=0
911

1012
num_models=${#MODEL_IDs[@]}

0 commit comments

Comments
 (0)