1 files changed, 1 insertions, 2 deletions

diff --git a/data/csv.py b/data/csv.py
index 8637ac1..253ce9e 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -68,13 +68,12 @@ class CSVDataModule(pl.LightningDataModule):
                 item.nprompt if "nprompt" in item else ""
             )
             for item in data
-            if "skip" not in item or item.skip != "x"
             for i in range(image_multiplier)
         ]
 
     def prepare_data(self):
         metadata = pd.read_csv(self.data_file)
-        metadata = list(metadata.itertuples())
+        metadata = [item for item in metadata.itertuples() if "skip" not in item or item.skip != "x"]
         num_images = len(metadata)
 
         valid_set_size = int(num_images * 0.2)