correct albumentation

try to make augmentation on GPU.
saved llm prompt

requirements.txt

@@ -3,12 +3,14 @@ torch
 torchvision
 torchaudio
 ultralytics
-albumentations~=2.0.4
+albumentations
 
-opencv-python~=4.11.0.86
-matplotlib~=3.10.0
-PyYAML~=6.0.2
-cryptography~=44.0.1
-numpy~=2.1.1
-requests~=2.32.3
-pyyaml
+opencv-python
+matplotlib
+PyYAML
+cryptography
+numpy
+requests
+pyyaml
+boto3
+nvidia-dali-cuda120