diff --git a/requirements-colab.txt b/requirements-colab.txt index facd9e1..641723a 100644 --- a/requirements-colab.txt +++ b/requirements-colab.txt @@ -9,6 +9,7 @@ loguru nltk<=3.8.1 num2words numpy<2 +onnx onnxruntime onnxruntime-gpu pyannote.audio>=3.1.0 diff --git a/requirements-infer.txt b/requirements-infer.txt index c198986..a54474e 100644 --- a/requirements-infer.txt +++ b/requirements-infer.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' 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'