feat(asr): add optional auxiliary timestamp model restoration for Canary (NVIDIA-NeMo#15268)

chaosido · AkCodes23 · commit 2d212b8f4c07 · 2026-01-28T16:49:09.000+05:30
Signed-off-by: chaosido &lt;wonnink.jesse@gmail.com&gt;
Signed-off-by: Akhil Varanasi &lt;akhilvaranasi23@gmail.com&gt;
diff --git a/nemo/collections/asr/models/aed_multitask_models.py b/nemo/collections/asr/models/aed_multitask_models.py
@@ -253,7 +253,10 @@ def __init__(self, cfg: DictConfig, trainer: Trainer = None):
         # Setup encoder adapters (from ASRAdapterModelMixin)
         self.setup_adapters()
 
-        timestamps_asr_model = self.__restore_timestamps_asr_model()
+        if self.cfg.get("restore_timestamps_model", True):
+            timestamps_asr_model = self.__restore_timestamps_asr_model()
+        else:
+            timestamps_asr_model = None
         # Using object.__setattr__ to bypass PyTorch's module registration
         object.__setattr__(self, 'timestamps_asr_model', timestamps_asr_model)
 
