Fix version of x-transformers lukas-blecher#33

0.15.1 included breaking changes
Also use ImageCompression instead of JpegCompression

Author: lukas-blecher

dataset/dataset.py

@@ -31,7 +31,7 @@
                      b_shift_limit=15, p=0.3),
         alb.GaussNoise(10, p=.2),
         alb.RandomBrightnessContrast(.05, (-.2, 0), True, p=0.2),
-        alb.JpegCompression(95, p=.3),
+        alb.ImageCompression(95, p=.3),
         alb.ToGray(always_apply=True),
         alb.Normalize((0.7931, 0.7931, 0.7931), (0.1738, 0.1738, 0.1738)),
         # alb.Sharpen()

requirements.txt

@@ -5,7 +5,7 @@ opencv_python_headless>=4.1.1.26
 requests>=2.22.0
 einops>=0.3.0
 chardet>=3.0.4
-x_transformers>=0.8.1
+x_transformers==0.15.0
 torch>=1.7.0
 imagesize>=1.2.0
 transformers>=4.2.2