requirements.txt 353 B

1234567891011121314151617
  1. torch==2.2.2
  2. torchvision==0.17.2
  3. flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
  4. timm==1.0.14
  5. onnx==1.14.0
  6. onnxruntime==1.15.1
  7. pycocotools==2.0.7
  8. PyYAML==6.0.1
  9. scipy==1.13.0
  10. onnxslim==0.1.31
  11. onnxruntime-gpu==1.18.0
  12. gradio==4.31.5
  13. opencv-python==4.9.0.80
  14. psutil==5.9.8
  15. py-cpuinfo==9.0.0
  16. huggingface-hub==0.23.2
  17. safetensors==0.4.3