WIP: parakeeet

2026-05-18 23:10:13 -03:00
parent 2e4aeb3b6f
commit aec56abe73
10 changed files with 549 additions and 654 deletions
--- a/stt_voice_messages/models/model_settings.py
+++ b/stt_voice_messages/models/model_settings.py
@@ -18,11 +18,6 @@ from dataclasses import dataclass, field


@dataclass
-class OpenAIWhisperSettings:
-    model_size: str = field(default='tiny', init=True)
-    translate_to_english: bool = field(default=False, init=True)
-
-@dataclass
-class FasterWhisperSettings:
-    model_size: str = field(default='tiny', init=True)
-    translate_to_english: bool = field(default=False, init=True)
+class OnnxAsrSettings:
+    model_id: str = field(default='nemo-parakeet-tdt-0.6b-v3', init=True)
+    model_path: str = ''