requirements.txt 457 B

1234567891011121314151617181920212223
  1. ninja
  2. psutil
  3. ray >= 2.9
  4. sentencepiece
  5. numpy
  6. torch >= 2.1.2
  7. transformers >= 4.36.0 # for mixtral
  8. uvicorn
  9. openai # for fastapi's openai proxy emulation
  10. xformers >= 0.0.23.post1
  11. einops # Required for phi-1_5
  12. fschat >= 0.2.23
  13. pydantic == 1.10.13
  14. fastapi
  15. colorlog
  16. einops # for phi
  17. aioprometheus[starlette] # for prometheus metrics
  18. triton >= 2.1.0
  19. lark == 1.1.8 # for grammars
  20. pynvml == 11.5.0
  21. gguf # for gguf
  22. scipy # for quip
  23. fast-hadamard-transform # for quip