田运杰 11 月之前
父節點
當前提交
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