requirements.txt 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. --extra-index-url https://download.pytorch.org/whl/cu121
  2. --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ # https://github.com/microsoft/onnxruntime/issues/21684
  3. conformer==0.3.2
  4. deepspeed==0.15.1; sys_platform == 'linux'
  5. diffusers==0.29.0
  6. fastapi==0.115.6
  7. fastapi-cli==0.0.4
  8. gdown==5.1.0
  9. gradio==5.4.0
  10. grpcio==1.57.0
  11. grpcio-tools==1.57.0
  12. hydra-core==1.3.2
  13. HyperPyYAML==1.2.2
  14. inflect==7.3.1
  15. librosa==0.10.2
  16. lightning==2.2.4
  17. matplotlib==3.7.5
  18. modelscope==1.20.0
  19. networkx==3.1
  20. omegaconf==2.3.0
  21. onnx==1.16.0
  22. onnxruntime-gpu==1.18.0; sys_platform == 'linux'
  23. onnxruntime==1.18.0; sys_platform == 'darwin' or sys_platform == 'win32'
  24. openai-whisper==20231117
  25. protobuf==4.25
  26. pyarrow==18.1.0
  27. pydantic==2.7.0
  28. pyworld==0.3.4
  29. rich==13.7.1
  30. soundfile==0.12.1
  31. tensorboard==2.14.0
  32. tensorrt-cu12==10.0.1; sys_platform == 'linux'
  33. tensorrt-cu12-bindings==10.0.1; sys_platform == 'linux'
  34. tensorrt-cu12-libs==10.0.1; sys_platform == 'linux'
  35. torch==2.3.1
  36. torchaudio==2.3.1
  37. transformers==4.51.3
  38. uvicorn==0.30.0
  39. wetext==0.0.4
  40. wget==3.2