Method assert_extra_args

train_textual_inversion.py:110–114

(self, args, train_dataset_group: Union[DatasetGroup, MinimalDataset], val_dataset_group: Optional[DatasetGroup])

        self.is_sdxl = False

    def assert_extra_args(self, args, train_dataset_group: Union[DatasetGroup, MinimalDataset], val_dataset_group: Optional[DatasetGroup]):
        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, _ = model_io.load_target_model(args, weight_dtype, accelerator)