From 225c533279dc5bff8bf11e38e61b256784746dfc Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Tue, 7 Mar 2023 08:23:34 +0900
Subject: [PATCH] accept empty caption  #258

---
 library/train_util.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/library/train_util.py b/library/train_util.py
index e15ce1336..8e22f65fa 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -924,7 +924,9 @@ def __init__(self, subsets: Sequence[FineTuningSubset], batch_size: int, tokeniz
         elif tags is not None and len(tags) > 0:
           caption = caption + ', ' + tags
           tags_list.append(tags)
-        assert caption is not None and len(caption) > 0, f"caption or tag is required / キャプションまたはタグは必須です:{abs_path}"
+
+        if caption is None:
+          caption = ""
 
         image_info = ImageInfo(image_key, subset.num_repeats, caption, False, abs_path)
         image_info.image_size = img_md.get('train_resolution')