lora_cluster / config.toml
BKM1804's picture
Upload config.toml with huggingface_hub
59fab41 verified
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