requirements.txt 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  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. numpy==1.26.4
  21. omegaconf==2.3.0
  22. onnx==1.16.0
  23. onnxruntime-gpu==1.18.0; sys_platform == 'linux'
  24. onnxruntime==1.18.0; sys_platform == 'darwin' or sys_platform == 'win32'
  25. openai-whisper==20231117
  26. protobuf==4.25
  27. pyarrow==18.1.0
  28. pydantic==2.7.0
  29. pyworld==0.3.4
  30. rich==13.7.1
  31. soundfile==0.12.1
  32. tensorboard==2.14.0
  33. tensorrt-cu12==10.13.3.9; sys_platform == 'linux'
  34. tensorrt-cu12-bindings==10.13.3.9; sys_platform == 'linux'
  35. tensorrt-cu12-libs==10.13.3.9; sys_platform == 'linux'
  36. torch==2.3.1
  37. torchaudio==2.3.1
  38. transformers==4.51.3
  39. x-transformers==2.11.24
  40. uvicorn==0.30.0
  41. wetext==0.0.4
  42. wget==3.2