Explorar o código

upload requirements

田运杰 hai 10 meses
pai
achega
f610fddfa7
Modificáronse 1 ficheiros con 17 adicións e 0 borrados
  1. 17 0
      requirements.txt

+ 17 - 0
requirements.txt

@@ -0,0 +1,17 @@
+torch==2.2.2 
+torchvision==0.17.2
+flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
+timm==1.0.14
+onnx==1.14.0
+onnxruntime==1.15.1
+pycocotools==2.0.7
+PyYAML==6.0.1
+scipy==1.13.0
+onnxslim==0.1.31
+onnxruntime-gpu==1.18.0
+gradio==4.31.5
+opencv-python==4.9.0.80
+psutil==5.9.8
+py-cpuinfo==9.0.0
+huggingface-hub==0.23.2
+safetensors==0.4.3