From ef3a110ae1fa49001c9ec7c699d3c2b1a99c5a76 Mon Sep 17 00:00:00 2001 From: rockerBOO Date: Sat, 5 Apr 2025 19:06:32 -0400 Subject: [PATCH] Undo train_network change --- train_network.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/train_network.py b/train_network.py index a7e747c6..f66cdeb4 100644 --- a/train_network.py +++ b/train_network.py @@ -171,10 +171,9 @@ class NetworkTrainer: train_dataset_group: Union[train_util.DatasetGroup, train_util.MinimalDataset], val_dataset_group: Optional[train_util.DatasetGroup], ): - if type(self).__name__ == "NetworkTrainer": - train_dataset_group.verify_bucket_reso_steps(64) - if val_dataset_group is not None: - val_dataset_group.verify_bucket_reso_steps(64) + train_dataset_group.verify_bucket_reso_steps(64) + if val_dataset_group is not None: + val_dataset_group.verify_bucket_reso_steps(64) def load_target_model(self, args, weight_dtype, accelerator): text_encoder, vae, unet, _ = train_util.load_target_model(args, weight_dtype, accelerator)