requirements.txt 1.1 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.27.2
  6. gdown==5.1.0
  7. gradio==5.4.0
  8. grpcio==1.57.0
  9. grpcio-tools==1.57.0
  10. huggingface-hub==0.25.2
  11. hydra-core==1.3.2
  12. HyperPyYAML==1.2.2
  13. inflect==7.3.1
  14. librosa==0.10.2
  15. lightning==2.2.4
  16. matplotlib==3.7.5
  17. modelscope==1.15.0
  18. networkx==3.1
  19. omegaconf==2.3.0
  20. onnx==1.16.0
  21. onnxruntime-gpu==1.18.0; sys_platform == 'linux'
  22. onnxruntime==1.18.0; sys_platform == 'darwin' or sys_platform == 'windows'
  23. openai-whisper==20231117
  24. protobuf==4.25
  25. pydantic==2.7.0
  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