File tree Expand file tree Collapse file tree 3 files changed +4
-4
lines changed
Expand file tree Collapse file tree 3 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 22
33export PYTHONPATH=" .." :$PYTHONPATH
44
5- MODEL_IDs=(" nvidia/canary-1b-flash " ) # options: "nvidia/canary-1b" "nvidia/canary-1b-flash" "nvidia/canary-180m-flash"
5+ MODEL_IDs=(" nvidia/canary-1b-v2 " ) # options: "nvidia/canary-1b" "nvidia/canary-1b-flash" "nvidia/canary-180m-flash" "nvidia/canary-1b-v2 "
66BATCH_SIZE=128
77DEVICE_ID=0
88
Original file line number Diff line number Diff line change 77export PYTHONPATH=" .." :$PYTHONPATH
88
99# Configuration
10- MODEL_ID=" nvidia/canary-1b-flash "
10+ MODEL_ID=" nvidia/canary-1b-v2 "
1111
1212BATCH_SIZE=64
1313
Original file line number Diff line number Diff line change 33export PYTHONPATH=" .." :$PYTHONPATH
44
55# considering latest model
6- MODEL_IDs=(" nvidia/parakeet-tdt-0.6b-v2 " )
6+ MODEL_IDs=(" nvidia/parakeet-tdt-0.6b-v3 " )
77
88# For RNNT models:
9- # ("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+ # ("nvidia/parakeet-tdt-0.6b-v2" "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")
1010
1111# For CTC models:
1212# ("nvidia/parakeet-ctc-1.1b" "nvidia/parakeet-ctc-0.6b" "nvidia/stt_en_fastconformer_ctc_large" "nvidia/stt_en_conformer_ctc_large" "nvidia/stt_en_conformer_ctc_small")
You can’t perform that action at this time.
0 commit comments