diff --git a/library/train_util.py b/library/train_util.py index 576fc5d8..55eeb316 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):