adjust min/max bucket reso divisible by reso steps #1632

2026-04-06 13:47:06 +00:00 · 2024-09-29 09:49:25 +09:00
parent ce49ced699
commit fe2aa32484
9 changed files with 48 additions and 8 deletions
--- a/fine_tune.py
+++ b/fine_tune.py
@@ -91,6 +91,8 @@ def train(args):
    ds_for_collator = train_dataset_group if args.max_data_loader_n_workers == 0 else None
    collator = train_util.collator_class(current_epoch, current_step, ds_for_collator)

+    train_dataset_group.verify_bucket_reso_steps(64)
+
    if args.debug_dataset:
        train_util.debug_dataset(train_dataset_group)
        return