updated NeMo version, batch size

KunalDhawan · KunalDhawan · commit 50aab948c623 · 2025-03-14T16:09:42.000-07:00
Signed-off-by: Kunal Dhawan &lt;kunaldhawan97@gmail.com&gt;
diff --git a/nemo_asr/run_canary.sh b/nemo_asr/run_canary.sh
@@ -3,7 +3,7 @@
 export PYTHONPATH="..":$PYTHONPATH
 
 MODEL_IDs=("nvidia/canary-1b-flash")  # options: "nvidia/canary-1b" "nvidia/canary-1b-flash" "nvidia/canary-180m-flash"
-BATCH_SIZE=64
+BATCH_SIZE=96
 DEVICE_ID=0
 
 num_models=${#MODEL_IDs[@]}
diff --git a/requirements/requirements_nemo.txt b/requirements/requirements_nemo.txt
@@ -1,4 +1,4 @@
-git+https://github.com/NVIDIA/NeMo.git@cb755f5595880a56159cc9a6f4a050c20a449d0a#egg=nemo_toolkit[asr] # This commit hash is a recent version of main at the time of testing.
+git+https://github.com/NVIDIA/NeMo.git@208e0da28e2ada8da84d8f7ddff8623efe1ff01c#egg=nemo_toolkit[asr] # This commit hash is a recent version of main at the time of testing.
 tqdm
 soundfile
 librosa

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-git+https://github.com/NVIDIA/NeMo.git@cb755f5595880a56159cc9a6f4a050c20a449d0a#egg=nemo_toolkit[asr] # This commit hash is a recent version of main at the time of testing.`
	`1`	`+git+https://github.com/NVIDIA/NeMo.git@208e0da28e2ada8da84d8f7ddff8623efe1ff01c#egg=nemo_toolkit[asr] # This commit hash is a recent version of main at the time of testing.`
`2`	`2`	`tqdm`
`3`	`3`	`soundfile`
`4`	`4`	`librosa`