requirements.txt 239 B

1234567891011121314151617
  1. ninja
  2. psutil
  3. ray ~= 2.5.1
  4. sentencepiece
  5. numpy
  6. torch ~= 2.0.0
  7. transformers >= 4.33.3
  8. uvicorn
  9. openai # for fastapi's openai proxy emulation
  10. xformers >= 0.0.22
  11. mypy
  12. pytest
  13. fschat >= 0.2.23
  14. pydantic < 2
  15. pyarrow # needed for ray
  16. pandas
  17. fastapi