Merge pull request #178 from tsukumijima/master

Fix: Missing onnx dependency
2024-11-17 12:35:17 +09:00
parent 875186f0c7 4a92a1abad
commit a6047012dd
3 changed files with 5 additions and 1 deletions
--- 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
--- 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
--- 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'