From d859a3a9259dd04d03a41816e17c8fd8bb0189ee Mon Sep 17 00:00:00 2001
From: TingTingin <36141041+TingTingin@users.noreply.github.com>
Date: Tue, 23 May 2023 02:00:33 -0400
Subject: [PATCH] Update train_util.py

fix mistake
---
 library/train_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/library/train_util.py b/library/train_util.py
index 05ec7f84..576fc5d8 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1071,7 +1071,7 @@ class DreamBoothDataset(BaseDataset):
                         assert len(lines) > 0, f"caption file is empty / キャプションファイルが空です: {cap_path}"
                         caption = lines[0].strip()
                     break
-                return caption
+             return caption
 
         def load_dreambooth_dir(subset: DreamBoothSubset):
             if not os.path.isdir(subset.image_dir):