train_batch_size = 1
gradient_checkpointing = true
network_train_unet_only = true
network_train_text_encoder_only = false
unet_lr = 0.0001
text_encoder_lr = 0.00001
lr_scheduler = "cosine_with_restarts"
lr_warmup_steps = 0
lr_scheduler_num_cycles = 2
optimizer_type = "AdamW8bit"
network_dim = 32
network_alpha = 16