File size: 5,021 Bytes
c180216 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 |
{
"training_method": "LORA",
"model_type": "STABLE_DIFFUSION_XL_10_BASE",
"debug_mode": false,
"debug_dir": "debug",
"workspace_dir": "workspace/run",
"cache_dir": "workspace-cache/run",
"tensorboard": true,
"tensorboard_expose": false,
"continue_last_backup": false,
"weight_dtype": "FLOAT_16",
"output_dtype": "FLOAT_32",
"output_model_format": "SAFETENSORS",
"gradient_checkpointing": true,
"circular_mask_generation": false,
"random_rotate_and_crop": false,
"aspect_ratio_bucketing": true,
"latent_caching": true,
"clear_cache_before_training": true,
"learning_rate_scheduler": "CONSTANT",
"learning_rate": 0.0003,
"learning_rate_warmup_steps": 200,
"learning_rate_cycles": 1,
"epochs": 20,
"batch_size": 4,
"gradient_accumulation_steps": 1,
"ema": "OFF",
"ema_decay": 0.999,
"ema_update_step_interval": 5,
"train_device": "cuda",
"temp_device": "cpu",
"train_dtype": "FLOAT_16",
"fallback_train_dtype": "BFLOAT_16",
"only_cache": false,
"resolution": "896",
"attention_mechanism": "XFORMERS",
"align_prop": false,
"align_prop_probability": 0.1,
"align_prop_loss": "AESTHETIC",
"align_prop_weight": 0.01,
"align_prop_steps": 20,
"align_prop_truncate_steps": 0.5,
"align_prop_cfg_scale": 7.0,
"mse_strength": 1.0,
"mae_strength": 0.0,
"vb_loss_strength": 1.0,
"loss_scaler": "NONE",
"learning_rate_scaler": "NONE",
"train_unet": true,
"train_unet_epochs": 100000,
"unet_learning_rate": null,
"offset_noise_weight": 0.0,
"perturbation_noise_weight": 0.0,
"rescale_noise_scheduler_to_zero_terminal_snr": false,
"force_v_prediction": false,
"force_epsilon_prediction": false,
"min_noising_strength": 0.0,
"max_noising_strength": 1.0,
"noising_weight": 0.0,
"noising_bias": 0.5,
"unet_weight_dtype": "NONE",
"train_prior": true,
"train_prior_epochs": 10000,
"prior_learning_rate": null,
"prior_weight_dtype": "NONE",
"train_text_encoder": false,
"train_text_encoder_epochs": 30,
"text_encoder_learning_rate": null,
"text_encoder_layer_skip": 0,
"text_encoder_weight_dtype": "NONE",
"train_text_encoder_2": false,
"train_text_encoder_2_epochs": 30,
"text_encoder_2_learning_rate": null,
"text_encoder_2_layer_skip": 0,
"text_encoder_2_weight_dtype": "NONE",
"vae_model_name": "",
"vae_weight_dtype": "FLOAT_32",
"effnet_encoder_model_name": "",
"effnet_encoder_weight_dtype": "NONE",
"decoder_model_name": "",
"decoder_weight_dtype": "NONE",
"decoder_text_encoder_weight_dtype": "NONE",
"decoder_vqgan_weight_dtype": "NONE",
"masked_training": false,
"unmasked_probability": 0.1,
"unmasked_weight": 0.1,
"normalize_masked_area_loss": false,
"embedding_model_names": [],
"token_count": 1,
"initial_embedding_text": "*",
"embedding_weight_dtype": "FLOAT_32",
"lora_model_name": "",
"lora_rank": 16,
"lora_alpha": 1.0,
"lora_weight_dtype": "FLOAT_32",
"optimizer": "ADAMW",
"optimizer_adam_w_mode": false,
"optimizer_alpha": null,
"optimizer_amsgrad": false,
"optimizer_beta1": 0.9,
"optimizer_beta2": 0.999,
"optimizer_beta3": null,
"optimizer_bias_correction": false,
"optimizer_block_wise": false,
"optimizer_capturable": false,
"optimizer_centered": false,
"optimizer_clip_threshold": null,
"optimizer_d0": null,
"optimizer_d_coef": null,
"optimizer_dampening": null,
"optimizer_decay_rate": null,
"optimizer_decouple": false,
"optimizer_differentiable": false,
"optimizer_eps": 1e-08,
"optimizer_eps2": null,
"optimizer_foreach": false,
"optimizer_fsdp_in_use": false,
"optimizer_fused": true,
"optimizer_growth_rate": null,
"optimizer_initial_accumulator_value": null,
"optimizer_is_paged": false,
"optimizer_log_every": null,
"optimizer_lr_decay": null,
"optimizer_max_unorm": null,
"optimizer_maximize": false,
"optimizer_min_8bit_size": null,
"optimizer_momentum": null,
"optimizer_nesterov": false,
"optimizer_no_prox": false,
"optimizer_optim_bits": null,
"optimizer_percentile_clipping": null,
"optimizer_relative_step": false,
"optimizer_safeguard_warmup": false,
"optimizer_scale_parameter": false,
"optimizer_use_bias_correction": false,
"optimizer_use_triton": false,
"optimizer_warmup_init": false,
"optimizer_weight_decay": 0.01,
"sample_definition_file_name": "training_samples/samples.json",
"sample_after": 10,
"sample_after_unit": "NEVER",
"sample_image_format": "JPG",
"samples_to_tensorboard": true,
"non_ema_sampling": true,
"backup_after": 10,
"backup_after_unit": "MINUTE",
"rolling_backup": false,
"rolling_backup_count": 3,
"backup_before_save": true,
"save_after": 2,
"save_after_unit": "EPOCH"
} |