田运杰 10 месяцев назад
Родитель
Сommit
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