From 360af27749f3123dd84bfdca0a8343f5c17fec91 Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Sun, 3 Sep 2023 12:27:58 +0900
Subject: [PATCH] fix ControlNetDataset not working

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

diff --git a/library/train_util.py b/library/train_util.py
index 200b486c..35bfb5f5 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1705,6 +1705,8 @@ class ControlNetDataset(BaseDataset):
                 subset.caption_dropout_rate,
                 subset.caption_dropout_every_n_epochs,
                 subset.caption_tag_dropout_rate,
+                subset.caption_prefix,
+                subset.caption_suffix,
                 subset.token_warmup_min,
                 subset.token_warmup_step,
             )