requirements

requirements.txt

@@ -2,6 +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
 onnx==1.14.0
 onnxruntime==1.15.1
 pycocotools==2.0.7
@@ -14,4 +15,4 @@ opencv-python==4.9.0.80
 psutil==5.9.8
 py-cpuinfo==9.0.0
 huggingface-hub==0.23.2
-safetensors==0.4.3
+safetensors==0.4.3