requirements.txt 543 B

123456789101112131415161718192021222324252627
  1. ninja
  2. psutil
  3. ray >= 2.9
  4. sentencepiece
  5. numpy
  6. torch == 2.2.0
  7. transformers == 4.39.0 # needed for cohere
  8. uvicorn
  9. openai # for fastapi's openai proxy emulation
  10. xformers >= 0.0.24
  11. einops # Required for phi-1_5
  12. fschat >= 0.2.23
  13. pydantic >= 2.0
  14. fastapi
  15. colorlog
  16. einops # for phi
  17. prometheus_client # for prometheus metrics
  18. triton >= 2.1.0
  19. lark == 1.1.8 # for grammars
  20. pynvml == 11.5.0
  21. scipy # for quip
  22. rich
  23. cupy-cuda12x == 12.3.0 # install cupy-cuda11x for CUDA 11.8
  24. outlines >= 0.0.27
  25. bitsandbytes >= 0.41.0
  26. loguru
  27. hf_transfer # for faster downloads