From 47ddc00c6a4c78c04ebf894f3075b3bb8ab40b63 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Mon, 20 Feb 2023 14:44:10 -0500 Subject: [PATCH] in textual inversion training, skip over non-image files - Closes #2715 --- ldm/invoke/training/textual_inversion_training.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ldm/invoke/training/textual_inversion_training.py b/ldm/invoke/training/textual_inversion_training.py index d65f4bf622..0c781519af 100644 --- a/ldm/invoke/training/textual_inversion_training.py +++ b/ldm/invoke/training/textual_inversion_training.py @@ -441,6 +441,7 @@ class TextualInversionDataset(Dataset): self.image_paths = [ os.path.join(self.data_root, file_path) for file_path in os.listdir(self.data_root) + if os.path.isfile(file_path) and file_path.endswith(('.png','.PNG','.jpg','.JPG','.jpeg','.JPEG','.gif','.GIF')) ] self.num_images = len(self.image_paths)