| { | |
| "adaptive_noise_scale": 0, | |
| "additional_parameters": "", | |
| "bucket_no_upscale": true, | |
| "bucket_reso_steps": 64, | |
| "cache_latents": true, | |
| "cache_latents_to_disk": false, | |
| "caption_dropout_every_n_epochs": 0.0, | |
| "caption_dropout_rate": 0.05, | |
| "caption_extension": "", | |
| "clip_skip": 2, | |
| "color_aug": false, | |
| "dataset_config": "./test/config/dataset.toml", | |
| "enable_bucket": true, | |
| "epoch": 1, | |
| "flip_aug": false, | |
| "full_bf16": false, | |
| "full_fp16": false, | |
| "gpu_ids": "", | |
| "gradient_accumulation_steps": 1, | |
| "gradient_checkpointing": false, | |
| "ip_noise_gamma": 0, | |
| "ip_noise_gamma_random_strength": false, | |
| "keep_tokens": "0", | |
| "learning_rate": 5e-05, | |
| "learning_rate_te": 1e-05, | |
| "learning_rate_te1": 1e-05, | |
| "learning_rate_te2": 1e-05, | |
| "log_tracker_config": "", | |
| "log_tracker_name": "", | |
| "logging_dir": "./test/logs", | |
| "lr_scheduler": "constant", | |
| "lr_scheduler_args": "", | |
| "lr_scheduler_num_cycles": "", | |
| "lr_scheduler_power": "", | |
| "lr_warmup": 0, | |
| "main_process_port": 12345, | |
| "masked_loss": false, | |
| "max_bucket_reso": 2048, | |
| "max_data_loader_n_workers": "0", | |
| "max_resolution": "512,512", | |
| "max_timestep": 1000, | |
| "max_token_length": "75", | |
| "max_train_epochs": "", | |
| "max_train_steps": "", | |
| "mem_eff_attn": false, | |
| "min_bucket_reso": 256, | |
| "min_snr_gamma": 0, | |
| "min_timestep": 0, | |
| "mixed_precision": "bf16", | |
| "model_list": "runwayml/stable-diffusion-v1-5", | |
| "multi_gpu": false, | |
| "multires_noise_discount": 0, | |
| "multires_noise_iterations": 0, | |
| "no_token_padding": false, | |
| "noise_offset": 0.05, | |
| "noise_offset_random_strength": false, | |
| "noise_offset_type": "Original", | |
| "num_cpu_threads_per_process": 2, | |
| "num_machines": 1, | |
| "num_processes": 1, | |
| "optimizer": "AdamW8bit", | |
| "optimizer_args": "", | |
| "output_dir": "./test/output", | |
| "output_name": "db-AdamW8bit-toml", | |
| "persistent_data_loader_workers": false, | |
| "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5", | |
| "prior_loss_weight": 1.0, | |
| "random_crop": false, | |
| "reg_data_dir": "", | |
| "resume": "", | |
| "sample_every_n_epochs": 0, | |
| "sample_every_n_steps": 25, | |
| "sample_prompts": "a painting of a gas mask , by darius kawasaki", | |
| "sample_sampler": "euler_a", | |
| "save_every_n_epochs": 1, | |
| "save_every_n_steps": 0, | |
| "save_last_n_steps": 0, | |
| "save_last_n_steps_state": 0, | |
| "save_model_as": "safetensors", | |
| "save_precision": "fp16", | |
| "save_state": false, | |
| "save_state_on_train_end": false, | |
| "scale_v_pred_loss_like_noise_pred": false, | |
| "sdxl": false, | |
| "seed": "1234", | |
| "shuffle_caption": false, | |
| "stop_text_encoder_training": 0, | |
| "train_batch_size": 4, | |
| "train_data_dir": "", | |
| "use_wandb": false, | |
| "v2": false, | |
| "v_parameterization": false, | |
| "v_pred_like_loss": 0, | |
| "vae": "", | |
| "vae_batch_size": 0, | |
| "wandb_api_key": "", | |
| "wandb_run_name": "", | |
| "weighted_captions": false, | |
| "xformers": "xformers" | |
| } |