tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Improve: Changed PyTorch to an optional dependency when used style_bert_vits2 as a library

Browse Source 
ONNXRuntime has relied on CUDA 12.x / cuDNN 9.x by default since v1.19.0, while PyTorch finally supports cuDNN 9 since v2.4.0.
Therefore, unless the pinning of torch<2.4 is eliminated, CUDA cannot be used with the latest version of ONNXRuntime.
This commit is contained in:
tsukumi
2024-09-17 17:56:59 +09:00
parent 5e2c83c6c7
commit b95420325b
4 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
requirements-infer.txt
View File

@@ -11,6 +11,7 @@ loguru
num2words
numpy<2
onnxruntime
onnxruntime-gpu
# onnxsim
# protobuf==4.25
psutil
@@ -22,7 +23,7 @@ pypinyin
pyworld-prebuilt
# stable_ts
# tensorboard
torch<2.4
torchaudio<2.4
torch
torchaudio
transformers
umap-learn