Enhanced whisper.py with MLX Whisper model support and added optional mlx-whisper to pyproject.toml. Added error handling for missing modules and created a new WhisperSTTServiceMLX class for MLX Whisper integration.

2025-03-16 02:18:54 -04:00
parent 8b86f6991d
commit 1baa52a17e
2 changed files with 150 additions and 11 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -65,6 +65,7 @@ langchain = [ "langchain~=0.3.14", "langchain-community~=0.3.14", "langchain-ope
 livekit = [ "livekit~=0.19.1", "livekit-api~=0.8.1", "tenacity~=9.0.0" ]
 lmnt = [ "websockets~=13.1" ]
 local = [ "pyaudio~=0.2.14" ]
+mlx-whisper = [ "mlx-whisper~=0.4.2" ]
 moondream = [ "einops~=0.8.0", "timm~=1.0.13", "transformers~=4.48.0" ]
 nim = []
 noisereduce = [ "noisereduce~=3.0.3" ]