requirements.txt 239 B

123456789101112131415161718
  1. -f https://download.pytorch.org/whl/torch_stable.html torch==2.0.1+cu118
  2. accelerate
  3. appdirs
  4. loralib
  5. bitsandbytes==0.39.1
  6. black
  7. black[jupyter]
  8. datasets
  9. fire
  10. peft
  11. transformers>=4.31.0
  12. sentencepiece
  13. py7zr
  14. scipy
  15. optimum
  16. vllm
  17. pytest
  18. pytest-mock