| { | |
| "adaptive_noise_scale": 0.005, | |
| "additional_parameters": "", | |
| "bucket_no_upscale": true, | |
| "bucket_reso_steps": 1, | |
| "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": "", | |
| "enable_bucket": true, | |
| "epoch": 8, | |
| "flip_aug": false, | |
| "full_fp16": false, | |
| "gpu_ids": "", | |
| "gradient_accumulation_steps": 1, | |
| "gradient_checkpointing": false, | |
| "init_word": "*", | |
| "ip_noise_gamma": 0.1, | |
| "ip_noise_gamma_random_strength": true, | |
| "keep_tokens": "0", | |
| "learning_rate": 0.0001, | |
| "log_tracker_config": "", | |
| "log_tracker_name": "", | |
| "logging_dir": "./test/logs", | |
| "lr_scheduler": "cosine", | |
| "lr_scheduler_args": "", | |
| "lr_scheduler_num_cycles": "", | |
| "lr_scheduler_power": "", | |
| "lr_warmup": 0, | |
| "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": 10, | |
| "min_timestep": 0, | |
| "mixed_precision": "bf16", | |
| "model_list": "runwayml/stable-diffusion-v1-5", | |
| "multi_gpu": false, | |
| "multires_noise_discount": 0.2, | |
| "multires_noise_iterations": 8, | |
| "no_token_padding": false, | |
| "noise_offset": 0.05, | |
| "noise_offset_random_strength": true, | |
| "noise_offset_type": "Original", | |
| "num_cpu_threads_per_process": 2, | |
| "num_machines": 1, | |
| "num_processes": 1, | |
| "num_vectors_per_token": 8, | |
| "optimizer": "AdamW8bit", | |
| "optimizer_args": "", | |
| "output_dir": "./test/output", | |
| "output_name": "TI-Adamw8bit", | |
| "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": 20, | |
| "sample_prompts": "a painting of man wearing 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, | |
| "scale_v_pred_loss_like_noise_pred": false, | |
| "sdxl": false, | |
| "sdxl_no_half_vae": false, | |
| "seed": "1234", | |
| "shuffle_caption": false, | |
| "stop_text_encoder_training": 0, | |
| "template": "style template", | |
| "token_string": "zxc", | |
| "train_batch_size": 4, | |
| "train_data_dir": "./test/img", | |
| "use_wandb": false, | |
| "v2": false, | |
| "v_parameterization": false, | |
| "v_pred_like_loss": 0, | |
| "vae": "", | |
| "vae_batch_size": 0, | |
| "wandb_api_key": "", | |
| "wandb_run_name": "", | |
| "weights": "", | |
| "xformers": "xformers" | |
| } |