diff --git a/library/train_util.py b/library/train_util.py index 4ac6728b..ba428e50 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -1774,6 +1774,7 @@ class ControlNetDataset(BaseDataset): tokenizer, max_token_length, resolution, + network_multiplier, enable_bucket, min_bucket_reso, max_bucket_reso,