requirements.txt 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839
  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.14.2; sys_platform == 'linux'
  5. diffusers==0.29.0
  6. gdown==5.1.0
  7. gradio==5.4.0
  8. grpcio==1.57.0
  9. grpcio-tools==1.57.0
  10. hydra-core==1.3.2
  11. HyperPyYAML==1.2.2
  12. inflect==7.3.1
  13. librosa==0.10.2
  14. lightning==2.2.4
  15. matplotlib==3.7.5
  16. modelscope==1.15.0
  17. networkx==3.1
  18. omegaconf==2.3.0
  19. onnx==1.16.0
  20. onnxruntime-gpu==1.18.0; sys_platform == 'linux'
  21. onnxruntime==1.18.0; sys_platform == 'darwin' or sys_platform == 'windows'
  22. openai-whisper==20231117
  23. protobuf==4.25
  24. pydantic==2.7.0
  25. pyworld==0.3.4
  26. rich==13.7.1
  27. soundfile==0.12.1
  28. tensorboard==2.14.0
  29. tensorrt-cu12==10.0.1; sys_platform == 'linux'
  30. tensorrt-cu12-bindings==10.0.1; sys_platform == 'linux'
  31. tensorrt-cu12-libs==10.0.1; sys_platform == 'linux'
  32. torch==2.3.1
  33. torchaudio==2.3.1
  34. transformers==4.40.1
  35. uvicorn==0.30.0
  36. wget==3.2
  37. fastapi==0.115.6
  38. fastapi-cli==0.0.4
  39. WeTextProcessing==1.0.3