VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix lllite training not working ref #913

Browse Source
This commit is contained in:
Kohya S
2023-12-10 09:23:37 +09:00
parent 42750f7846
commit 4a2cef887c
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
library/train_util.py
View File

@@ -1722,6 +1722,7 @@ class ControlNetDataset(BaseDataset):
subset.caption_extension, subset.caption_extension,
subset.num_repeats, subset.num_repeats,
subset.shuffle_caption, subset.shuffle_caption,
subset.caption_separator,
subset.keep_tokens, subset.keep_tokens,
subset.color_aug, subset.color_aug,
subset.flip_aug, subset.flip_aug,