Update train_util.py

2026-04-09 06:45:09 +00:00 · 2023-05-23 02:02:39 -04:00
parent d859a3a925
commit 061e157191
1 changed files with 1 additions and 1 deletions
--- 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):