田运杰 hai 1 ano
pai
achega
ed2b434655
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      requirements.txt

+ 1 - 1
requirements.txt

@@ -2,7 +2,7 @@ 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
-albumentations
+albumentations==2.0.4
 onnx==1.14.0
 onnxruntime==1.15.1
 pycocotools==2.0.7