bucket_no_upscale = true | |
bucket_reso_steps = 64 | |
cache_latents = true | |
caption_extension = ".txt" | |
clip_skip = 1 | |
dynamo_backend = "no" | |
enable_bucket = true | |
epoch = 20 | |
gradient_accumulation_steps = 1 | |
huber_c = 0.1 | |
huber_schedule = "snr" | |
huggingface_repo_id = "BKM1804/lora_cluster" | |
huggingface_repo_type = "model" | |
huggingface_repo_visibility = "public" | |
learning_rate = 0.0001 | |
loss_type = "l2" | |
lr_scheduler = "cosine_with_restarts" | |
lr_scheduler_args = [] | |
lr_scheduler_num_cycles = 1 | |
lr_scheduler_power = 1 | |
lr_warmup_steps = 300 | |
max_bucket_reso = 2048 | |
max_data_loader_n_workers = 0 | |
max_grad_norm = 1 | |
max_timestep = 1000 | |
max_token_length = 75 | |
max_train_steps = 3000 | |
min_bucket_reso = 256 | |
min_snr_gamma = 5 | |
mixed_precision = "fp16" | |
multires_noise_discount = 0.3 | |
network_alpha = 128 | |
network_args = [] | |
network_dim = 128 | |
network_module = "networks.lora" | |
no_half_vae = true | |
noise_offset = 0.1 | |
noise_offset_type = "Original" | |
optimizer_args = [] | |
optimizer_type = "AdamW" | |
output_dir = "/workspace/kohya_ss/outputs" | |
output_name = "last" | |
pretrained_model_name_or_path = "stabilityai/stable-diffusion-xl-base-1.0" | |
prior_loss_weight = 1 | |
resolution = "512" | |
sample_prompts = "/workspace/kohya_ss/outputs/prompt.txt" | |
sample_sampler = "euler_a" | |
save_every_n_epochs = 1 | |
save_model_as = "safetensors" | |
save_precision = "fp16" | |
text_encoder_lr = 4e-5 | |
train_batch_size = 1 | |
train_data_dir = "/workspace/data/img" | |
unet_lr = 0.0001 | |
xformers = true | |