From 4a2cef887c9b7546ab463b276e65ffdcb478323e Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Sun, 10 Dec 2023 09:23:37 +0900
Subject: [PATCH] fix lllite training not working ref #913

---
 library/train_util.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/library/train_util.py b/library/train_util.py
index 13a07b3c..d2eb7cb2 100644
--- 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,