diff --git a/setup.py b/setup.py index ecc040cc..bc00ff1f 100644 --- a/setup.py +++ b/setup.py @@ -45,10 +45,14 @@ "torch", "torchaudio", "websockets", - "onnxruntime", + "onnxruntime==1.16.0", "ffmpeg-python", "scipy", "websocket-client", + "numba", + "openai-whisper", + "kaldialign", + "soundfile", ], python_requires=">=3.8" )