BatonVoice / requirements.txt
sunnyzjx's picture
Update requirements.txt
6d2d2b9 verified
--extra-index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ # https://github.com/microsoft/onnxruntime/issues/21684
conformer
deepspeed
diffusers==0.29.0
fastapi
fastapi-cli
gdown==5.1.0
gradio==3.43.2
grpcio
grpcio-tools
hydra-core==1.3.2
HyperPyYAML==1.2.2
inflect==7.3.1
librosa==0.10.2
lightning
matplotlib==3.7.5
modelscope==1.20.0
networkx==3.1
omegaconf==2.3.0
onnx==1.16.0
onnxruntime-gpu==1.18.0; sys_platform == 'linux'
onnxruntime==1.18.0; sys_platform == 'darwin' or sys_platform == 'win32'
openai-whisper
praat-parselmouth
protobuf==4.25
pyarrow==18.1.0
pyworld==0.3.4
rich==13.7.1
soundfile==0.12.1
tensorboard==2.14.0
tensorrt-cu12==10.0.1; sys_platform == 'linux'
tensorrt-cu12-bindings==10.0.1; sys_platform == 'linux'
tensorrt-cu12-libs==10.0.1; sys_platform == 'linux'
torch==2.7.1
torchaudio
transformers==4.56.2
triton
uvicorn==0.30.0
vllm==0.10.1
wetext==0.0.4
wget==3.2
modelscope