田运杰 пре 10 месеци
родитељ
комит
ed2b434655
1 измењених фајлова са 1 додато и 1 уклоњено
  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