diff --git a/requirements.txt b/requirements.txt
index 39c8c1a..36ce7ce 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ sentence_transformers
 pyyaml
 tqdm
 numpy
-easydict==1.9.0
+easydict
 scikit-image
 scikit-learn
 opencv-python
@@ -11,13 +11,13 @@ tensorflow
 joblib
 matplotlib
 pandas
-albumentations==0.5.2
-hydra-core==1.1.0
-pytorch-lightning==1.2.9
+albumentations
+hydra-core
 tabulate
-kornia==0.5.0
+kornia
 webdataset
 packaging
 wldhx.yadisk-direct
 timm
-moviepy
\ No newline at end of file
+moviepy
+tf-keras
\ No newline at end of file