1
0

requirements.txt 390 B

1234567891011121314151617181920
  1. ninja
  2. psutil
  3. ray ~= 2.5.1
  4. sentencepiece
  5. numpy
  6. torch ~= 2.0.1
  7. transformers >= 4.36.0 # for mixtral
  8. uvicorn
  9. openai # for fastapi's openai proxy emulation
  10. xformers >= 0.0.22
  11. einops # Required for phi-1_5
  12. fschat >= 0.2.23
  13. pydantic == 1.10.13
  14. pyarrow # needed for ray
  15. pandas
  16. fastapi
  17. colorlog
  18. einops # for phi
  19. aioprometheus[starlette] # for prometheus metrics
  20. pillow # image processing for llava