requirements-rocm.txt 393 B

123456789101112131415
  1. ninja # For faster builds.
  2. typing-extensions>=4.8.0
  3. psutil
  4. ray >= 2.5.1
  5. pandas # Required for Ray data.
  6. pyarrow # Required for Ray data.
  7. sentencepiece # Required for LLaMA tokenizer.
  8. numpy
  9. tokenizers>=0.15.0
  10. huggingface_hub<0.18,>=0.16.4
  11. einops # Required for phi-1_5
  12. transformers >= 4.34.0 # Required for Mistral.
  13. fastapi
  14. uvicorn[standard]
  15. pydantic >= 2.0 # Required for OpenAI server.