Model_Compilation / Training_model_data /training_config.toml
Coercer's picture
Upload 2 files
55faa5a verified
[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"