Add: Preparation for ONNX inference support ①

2024-09-17 06:53:11 +09:00
parent a037b955ac
commit eede796c4e
4 changed files with 141 additions and 37 deletions
--- a/requirements-infer.txt
+++ b/requirements-infer.txt
@@ -4,12 +4,14 @@ cn2an
 # faster-whisper==0.10.1
 g2p_en
 GPUtil
-gradio
+gradio>=4.32
 jieba
 # librosa==0.9.2
 loguru
 num2words
 numpy<2
+onnxruntime
+# onnxsim
 # protobuf==4.25
 psutil
 # punctuators
@@ -21,5 +23,6 @@ pyworld-prebuilt
 # stable_ts
 # tensorboard
 torch<2.4
+torchaudio<2.4
 transformers
 umap-learn