feat(dictate): add voice-reactive listening waveform

2026-08-26 13:40:47 -04:00
parent e234468a73
commit d1a811a5c5
16 changed files with 383 additions and 73 deletions
@@ -86,12 +86,13 @@ class NemoParakeetModel:
        os.environ["NEMO_OFFLINE"] = "1"

        import numpy
-        from nemo.collections.asr.models import ASRModel
+        import torch
+        from nemo.collections.asr.models import EncDecRNNTBPEModel
        from omegaconf import open_dict

-        model = ASRModel.restore_from(
+        model = EncDecRNNTBPEModel.restore_from(
            restore_path=str(self._model_path),
-            map_location="cuda",
+            map_location=torch.device("cuda"),
        )
        model.to("cuda")
        model.eval()