fix lllite training not working ref #913

kohya-ss · Dec 10, 2023 · 4a2cef8 · 4a2cef8
1 parent 42750f7
commit 4a2cef8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/library/train_util.py b/library/train_util.py
@@ -1722,6 +1722,7 @@ def __init__(
                 subset.caption_extension,
                 subset.num_repeats,
                 subset.shuffle_caption,
+                subset.caption_separator,
                 subset.keep_tokens,
                 subset.color_aug,
                 subset.flip_aug,