1 files changed, 2 insertions, 2 deletions

diff --git a/data/csv.py b/data/csv.py
index e25dd3f..edce2b1 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -49,7 +49,7 @@ class CSVDataModule():
             data_file: str,
             prompt_processor: PromptProcessor,
             class_subdir: str = "cls",
-            num_class_images: int = 100,
+            num_class_images: int = 1,
             size: int = 512,
             repeats: int = 1,
             dropout: float = 0,
@@ -117,7 +117,7 @@ class CSVDataModule():
         return [item for item in items if self.filter(item)]
 
     def pad_items(self, items: list[CSVDataItem], num_class_images: int = 1) -> list[CSVDataItem]:
-        image_multiplier = max(math.ceil(num_class_images / len(items)), 1)
+        image_multiplier = max(num_class_images, 1)
 
         return [
             CSVDataItem(