fix lllite training not working ref #913

2026-04-08 22:35:09 +00:00 · 2023-12-10 09:23:37 +09:00
parent 42750f7846
commit 4a2cef887c
1 changed files with 1 additions and 0 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1722,6 +1722,7 @@ class ControlNetDataset(BaseDataset):
                subset.caption_extension,
                subset.num_repeats,
                subset.shuffle_caption,
+                subset.caption_separator,
                subset.keep_tokens,
                subset.color_aug,
                subset.flip_aug,