diff --git a/magvit2_pytorch/data.py b/magvit2_pytorch/data.py
index ab0ed3a..b9ddd3d 100644
--- a/magvit2_pytorch/data.py
+++ b/magvit2_pytorch/data.py
@@ -86,6 +86,8 @@ def __init__(
         self.folder = folder
 
         self.image_size = image_size
+
+        exts = exts + [ext.upper() for ext in exts]
         self.paths = [p for ext in exts for p in folder.glob(f'**/*.{ext}')]
 
         print(f'{len(self.paths)} training samples found at {folder}')
diff --git a/magvit2_pytorch/version.py b/magvit2_pytorch/version.py
index a1a111f..a33fc87 100644
--- a/magvit2_pytorch/version.py
+++ b/magvit2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.1.44'
+__version__ = '0.1.45'