Merge pull request #17 from nicolai256/main

modified the dataloader for PIL errors and added progress bar

ldm/data/data_loader.py

@@ -1,8 +1,12 @@
 import os
 from PIL import Image
+import PIL
 import random
 from ldm.data.image_train_item import ImageTrainItem
 import ldm.data.aspects as aspects
+from tqdm import tqdm
+
+PIL.Image.MAX_IMAGE_PIXELS = 933120000
         
 class DataLoaderMultiAspect():
     """
@@ -50,7 +54,7 @@ class DataLoaderMultiAspect():
         """
         decorated_image_train_items = []
 
-        for pathname in image_paths:
+        for pathname in tqdm(image_paths):
             caption_from_filename = os.path.splitext(os.path.basename(pathname))[0].split("_")[0]
 
             txt_file_path = os.path.splitext(pathname)[0] + ".txt"