Upload 2 files

Training_model_data/dataset_config.toml

@@ -0,0 +1,17 @@
+[[datasets]]
+
+[[datasets.subsets]]
+num_repeats = 1
+image_dir = "/content/drive/MyDrive/Loras/Raichu/dataset"
+
+[general]
+resolution = 1024
+shuffle_caption = true
+keep_tokens = 1
+flip_aug = false
+caption_extension = ".txt"
+enable_bucket = true
+bucket_no_upscale = false
+bucket_reso_steps = 64
+min_bucket_reso = 256
+max_bucket_reso = 4096

Training_model_data/training_config.toml

@@ -0,0 +1,54 @@
+[model_arguments]
+unet_lr = 8e-7
+text_encoder_lr = 0
+v2 = false
+sdxl = true
+network_train_unet_only = true
+
+[optimizer_arguments]
+learning_rate = 8e-7
+lr_scheduler = "cosine_with_restarts"
+lr_scheduler_num_cycles = 3
+lr_warmup_steps = 174
+optimizer_type = "AdaFactor"
+optimizer_args = [ "scale_parameter=False", "relative_step=False", "warmup_init=False",]
+loss_type = "l2"
+max_grad_norm = 1.0
+
+[training_arguments]
+lowram = true
+pretrained_model_name_or_path = "/content/novaPKMXL_v10.safetensors"
+vae = "/content/sdxl_vae.safetensors"
+max_train_epochs = 10
+train_batch_size = 1
+seed = 42
+max_token_length = 75
+xformers = true
+sdpa = false
+min_snr_gamma = 5
+no_half_vae = false
+gradient_checkpointing = true
+gradient_accumulation_steps = 1
+max_data_loader_n_workers = 1
+persistent_data_loader_workers = false
+mixed_precision = "fp16"
+full_fp16 = true
+full_bf16 = false
+cache_latents = true
+cache_latents_to_disk = true
+cache_text_encoder_outputs = false
+min_timestep = 0
+max_timestep = 1000
+prior_loss_weight = 1.0
+multires_noise_iterations = 0
+multires_noise_discount = 0.3
+
+[saving_arguments]
+save_precision = "fp16"
+save_model_as = "ckpt"
+save_every_n_epochs = 1
+save_last_n_epochs = 1
+output_name = "Raichu"
+output_dir = "/content/drive/MyDrive/Loras/Raichu/output"
+log_prefix = "Raichu"
+logging_dir = "/content/drive/MyDrive/Loras/_logs"