diff --git a/requirements-colab.txt b/requirements-colab.txt index facd9e1..0664647 100644 --- a/requirements-colab.txt +++ b/requirements-colab.txt @@ -9,8 +9,10 @@ loguru nltk<=3.8.1 num2words numpy<2 +onnx onnxruntime onnxruntime-gpu +onnxsim pyannote.audio>=3.1.0 pyloudnorm pyopenjtalk-dict diff --git a/requirements-infer.txt b/requirements-infer.txt index c198986..f856be7 100644 --- a/requirements-infer.txt +++ b/requirements-infer.txt @@ -11,10 +11,11 @@ loguru nltk<=3.8.1 num2words numpy<2 +onnx onnxruntime onnxruntime-directml; sys_platform == 'win32' onnxruntime-gpu; sys_platform != 'darwin' -# onnxsim +onnxsim # protobuf==4.25 psutil # punctuators diff --git a/requirements.txt b/requirements.txt index dd2d327..360de3b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,6 +11,7 @@ loguru nltk<=3.8.1 num2words numpy<2 +onnx onnxruntime onnxruntime-directml; sys_platform == 'win32' onnxruntime-gpu; sys_platform != 'darwin'