田运杰 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