From f5b656d21c5b449eed6ce212e909043c124f79ee Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 12 Oct 2022 08:18:22 +0200 Subject: Various updates --- data/csv.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'data') 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) -- cgit v1.2.3-70-g09d2