fix ControlNetDataset not working

2026-04-09 06:45:09 +00:00 · 2023-09-03 12:27:58 +09:00
parent 0ee75fd75d
commit 360af27749
1 changed files with 2 additions and 0 deletions
--- 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,
            )