Trained on images generated with CAT - Citron Anime Treasure with the goal to copy this style to PonyXL

--

Trained on: 120 images

Steps: 434/620

Epoch: 7/10

Time for Epoch 7: 1:00:47

Colab Trainer crashed at:

epoch 8/10

steps: 77% 480/620 [1:07:15<19:37]

--

[network_arguments]

unet_lr = 0.75

text_encoder_lr = 0.75

network_dim = 32

network_alpha = 32

network_module = "networks.lora"

network_train_unet_only = false

[optimizer_arguments]

learning_rate = 0.75

lr_scheduler = "cosine_with_restarts"

lr_scheduler_num_cycles = 3

lr_warmup_steps = 30

optimizer_type = "Prodigy"

optimizer_args = [ "weight_decay=0.1", "betas=[0.9,0.99]",]

[training_arguments]

pretrained_model_name_or_path = "stablediffusionapi/pony-diffusion-v6-xl"

vae = "stabilityai/sdxl-vae"

max_train_epochs = 10

train_batch_size = 4

seed = 42

max_token_length = 225

xformers = false

sdpa = true

min_snr_gamma = 8.0

lowram = true

no_half_vae = true

gradient_checkpointing = true

gradient_accumulation_steps = 1

max_data_loader_n_workers = 8

persistent_data_loader_workers = true

mixed_precision = "fp16"

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

[saving_arguments]

save_precision = "fp16"

save_model_as = "safetensors"

save_every_n_epochs = 1

save_last_n_epochs = 10

output_name = "CitronAnimeTreasure"

output_dir = "/content/drive/MyDrive/lora_training/output/CitronAnimeTreasure"

log_prefix = "CitronAnimeTreasure"

logging_dir = "/content/drive/MyDrive/lora_training/log"