requirements-rocm.txt 336 B

1234567891011121314
  1. ninja # For faster builds.
  2. typing-extensions>=4.8.0
  3. starlette
  4. psutil
  5. ray >= 2.9
  6. sentencepiece # Required for LLaMA tokenizer.
  7. numpy
  8. tokenizers>=0.15.0
  9. einops # Required for phi-1_5
  10. transformers >= 4.39.0 # Required for Mistral.
  11. fastapi
  12. uvicorn[standard]
  13. pydantic == 1.10.13 # Required for OpenAI server.
  14. prometheus_client >= 0.18.0