take care of all caps image extensions, for lucidrains/magvit2-pytorc…

…h#19
trustme000777 · Dec 7, 2023 · 718723a · 718723a
1 parent d00ed09
commit 718723a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -1 +1 @@
-__version__ = '0.1.44'
+__version__ = '0.1.45'