From 36440e48ce279872d6e736bcb1bf57d13da73a11 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Mon, 16 Jan 2023 17:09:01 +0100
Subject: Moved multi-TI code from Dreambooth to TI script

---
 data/csv.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'data')

diff --git a/data/csv.py b/data/csv.py
index 85b98f8..6857b6f 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -282,7 +282,7 @@ class VlpnDataModule():
         collate_fn_ = partial(collate_fn, self.dtype, self.tokenizer, self.num_class_images != 0)
 
         if valid_set_size == 0:
-            data_train, data_val = items, items[:1]
+            data_train, data_val = items, []
         else:
             data_train, data_val = random_split(items, [train_set_size, valid_set_size], generator=generator)
 
-- 
cgit v1.2.3-70-g09d2