| bucket_no_upscale = true | |
| bucket_reso_steps = 64 | |
| cache_latents = true | |
| cache_latents_to_disk = true | |
| caption_extension = ".simple" | |
| clip_skip = 1 | |
| dynamo_backend = "no" | |
| enable_bucket = true | |
| epoch = 20 | |
| flip_aug = true | |
| full_bf16 = true | |
| gradient_accumulation_steps = 1 | |
| gradient_checkpointing = true | |
| huber_c = 0.1 | |
| huber_schedule = "snr" | |
| learning_rate = 2e-6 | |
| learning_rate_te1 = 1e-6 | |
| learning_rate_te2 = 1e-6 | |
| logging_dir = "/workspace/logs" | |
| loss_type = "l2" | |
| lr_scheduler = "constant" | |
| lr_scheduler_args = [] | |
| lr_scheduler_num_cycles = 1 | |
| lr_scheduler_power = 1 | |
| max_bucket_reso = 2048 | |
| max_data_loader_n_workers = 0 | |
| max_timestep = 1000 | |
| max_token_length = 75 | |
| max_train_steps = 59795 | |
| min_bucket_reso = 64 | |
| mixed_precision = "bf16" | |
| noise_offset_type = "Original" | |
| optimizer_args = [ "scale_parameter=False", "relative_step=False", "warmup_init=False",] | |
| optimizer_type = "Adafactor" | |
| output_dir = "/workspace/train/models" | |
| output_name = "SDXL_20240528" | |
| pretrained_model_name_or_path = "/workspace/train/models/SDXL_20240527-000003.safetensors" | |
| prior_loss_weight = 1 | |
| reg_data_dir = "/workspace/train/datasets/main" | |
| resolution = "1024,1024" | |
| sample_every_n_steps = 500 | |
| sample_prompts = "/workspace/train/models/prompt.txt" | |
| sample_sampler = "euler_a" | |
| save_every_n_epochs = 1 | |
| save_model_as = "safetensors" | |
| save_precision = "bf16" | |
| train_batch_size = 8 | |
| train_data_dir = "/workspace/train/datasets/celebs" | |
| vae_batch_size = 6 | |
| xformers = true | |