requirements-common.txt 606 B

12345678910111213141516171819202122232425262728
  1. cmake >= 3.21
  2. ninja
  3. psutil
  4. sentencepiece
  5. numpy < 2.0.0
  6. transformers == 4.43.0 # needed for llama
  7. uvicorn
  8. openai # for fastapi's openai proxy emulation
  9. einops # Required for phi-1_5
  10. fschat >= 0.2.23
  11. pydantic >= 2.0
  12. fastapi
  13. colorlog
  14. einops # for phi
  15. prometheus_client # for prometheus metrics
  16. prometheus-fastapi-instrumentator >= 7.0.0
  17. lark == 1.1.8 # for grammars
  18. scipy # for quip
  19. rich
  20. loguru
  21. hf_transfer # for faster downloads
  22. sentence-transformers # for embeddings
  23. tiktoken >= 0.6.0 # for DBRX tokenizer
  24. outlines >= 0.0.43, < 0.1 # Requires torch >= 2.1.0
  25. lm-format-enforcer >= 0.10.3
  26. uvloop
  27. pillow
  28. pyzmq