requirements-rocm.txt 409 B

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