| { | |
| "__version": 3, | |
| "training_method": "FINE_TUNE", | |
| "model_type": "STABLE_DIFFUSION_XL_10_BASE", | |
| "debug_mode": false, | |
| "debug_dir": "debug", | |
| "workspace_dir": "/home/Ubuntu/Desktop/onetrainer_workspace", | |
| "cache_dir": "/home/Ubuntu/Desktop/onetrainer_workspace/cache", | |
| "tensorboard": false, | |
| "tensorboard_expose": false, | |
| "continue_last_backup": false, | |
| "include_train_config": "NONE", | |
| "base_model_name": "/home/Ubuntu/Desktop/1T/juggernautXL_juggernautX.safetensors", | |
| "weight_dtype": "BFLOAT_16", | |
| "output_dtype": "BFLOAT_16", | |
| "output_model_format": "SAFETENSORS", | |
| "output_model_destination": "/home/Ubuntu/Desktop/onetrainer_workspace/brkcat.safetensors", | |
| "gradient_checkpointing": true, | |
| "force_circular_padding": false, | |
| "concept_file_name": "training_concepts/concepts.json", | |
| "concepts": [ | |
| { | |
| "__version": 1, | |
| "image": { | |
| "__version": 0, | |
| "enable_crop_jitter": false, | |
| "enable_random_flip": false, | |
| "enable_fixed_flip": false, | |
| "enable_random_rotate": false, | |
| "enable_fixed_rotate": false, | |
| "random_rotate_max_angle": 0.0, | |
| "enable_random_brightness": false, | |
| "enable_fixed_brightness": false, | |
| "random_brightness_max_strength": 0.0, | |
| "enable_random_contrast": false, | |
| "enable_fixed_contrast": false, | |
| "random_contrast_max_strength": 0.0, | |
| "enable_random_saturation": false, | |
| "enable_fixed_saturation": false, | |
| "random_saturation_max_strength": 0.0, | |
| "enable_random_hue": false, | |
| "enable_fixed_hue": false, | |
| "random_hue_max_strength": 0.0, | |
| "enable_resolution_override": false, | |
| "resolution_override": "512" | |
| }, | |
| "text": { | |
| "__version": 0, | |
| "prompt_source": "concept", | |
| "prompt_path": "/home/Ubuntu/Desktop/1T/brk cat.txt", | |
| "enable_tag_shuffling": false, | |
| "tag_delimiter": ",", | |
| "keep_tags_count": 1 | |
| }, | |
| "name": "training", | |
| "path": "/home/Ubuntu/Desktop/1T/brk1024", | |
| "seed": 866513600, | |
| "enabled": true, | |
| "include_subdirectories": false, | |
| "image_variations": 1, | |
| "text_variations": 1, | |
| "balancing": 1.0, | |
| "balancing_strategy": "REPEATS", | |
| "loss_weight": 1.0 | |
| }, | |
| { | |
| "__version": 1, | |
| "image": { | |
| "__version": 0, | |
| "enable_crop_jitter": false, | |
| "enable_random_flip": false, | |
| "enable_fixed_flip": false, | |
| "enable_random_rotate": false, | |
| "enable_fixed_rotate": false, | |
| "random_rotate_max_angle": 0.0, | |
| "enable_random_brightness": false, | |
| "enable_fixed_brightness": false, | |
| "random_brightness_max_strength": 0.0, | |
| "enable_random_contrast": false, | |
| "enable_fixed_contrast": false, | |
| "random_contrast_max_strength": 0.0, | |
| "enable_random_saturation": false, | |
| "enable_fixed_saturation": false, | |
| "random_saturation_max_strength": 0.0, | |
| "enable_random_hue": false, | |
| "enable_fixed_hue": false, | |
| "random_hue_max_strength": 0.0, | |
| "enable_resolution_override": false, | |
| "resolution_override": "512" | |
| }, | |
| "text": { | |
| "__version": 0, | |
| "prompt_source": "concept", | |
| "prompt_path": "/home/Ubuntu/Desktop/1T/cat.txt", | |
| "enable_tag_shuffling": false, | |
| "tag_delimiter": ",", | |
| "keep_tags_count": 1 | |
| }, | |
| "name": "reg", | |
| "path": "/home/Ubuntu/Desktop/1T/cat-unsplash1024x1024", | |
| "seed": 920357731, | |
| "enabled": true, | |
| "include_subdirectories": false, | |
| "image_variations": 1, | |
| "text_variations": 1, | |
| "balancing": 0.02, | |
| "balancing_strategy": "REPEATS", | |
| "loss_weight": 1.0 | |
| } | |
| ], | |
| "circular_mask_generation": false, | |
| "random_rotate_and_crop": false, | |
| "aspect_ratio_bucketing": false, | |
| "latent_caching": true, | |
| "clear_cache_before_training": false, | |
| "learning_rate_scheduler": "CONSTANT", | |
| "learning_rate": 1e-05, | |
| "learning_rate_warmup_steps": 200, | |
| "learning_rate_cycles": 1, | |
| "epochs": 200, | |
| "batch_size": 1, | |
| "gradient_accumulation_steps": 1, | |
| "ema": "OFF", | |
| "ema_decay": 0.999, | |
| "ema_update_step_interval": 1, | |
| "dataloader_threads": 2, | |
| "train_device": "cuda", | |
| "temp_device": "cpu", | |
| "train_dtype": "BFLOAT_16", | |
| "fallback_train_dtype": "FLOAT_32", | |
| "enable_autocast_cache": true, | |
| "only_cache": false, | |
| "resolution": "1024", | |
| "attention_mechanism": "DEFAULT", | |
| "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_weight_fn": "CONSTANT", | |
| "loss_weight_strength": 5.0, | |
| "dropout_probability": 0.0, | |
| "loss_scaler": "NONE", | |
| "learning_rate_scaler": "NONE", | |
| "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": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": true, | |
| "stop_training_after": 10000, | |
| "stop_training_after_unit": "EPOCH", | |
| "learning_rate": 1e-05, | |
| "weight_dtype": "BFLOAT_16" | |
| }, | |
| "prior": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": true, | |
| "stop_training_after": 10000, | |
| "stop_training_after_unit": "EPOCH", | |
| "learning_rate": null, | |
| "weight_dtype": "NONE" | |
| }, | |
| "text_encoder": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": true, | |
| "stop_training_after": 250, | |
| "stop_training_after_unit": "EPOCH", | |
| "learning_rate": 3e-06, | |
| "weight_dtype": "BFLOAT_16" | |
| }, | |
| "text_encoder_layer_skip": 0, | |
| "text_encoder_2": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": false, | |
| "stop_training_after": 30, | |
| "stop_training_after_unit": "EPOCH", | |
| "learning_rate": null, | |
| "weight_dtype": "BFLOAT_16" | |
| }, | |
| "text_encoder_2_layer_skip": 0, | |
| "vae": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": true, | |
| "stop_training_after": null, | |
| "stop_training_after_unit": "NEVER", | |
| "learning_rate": null, | |
| "weight_dtype": "FLOAT_32" | |
| }, | |
| "effnet_encoder": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": true, | |
| "stop_training_after": null, | |
| "stop_training_after_unit": "NEVER", | |
| "learning_rate": null, | |
| "weight_dtype": "NONE" | |
| }, | |
| "decoder": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": true, | |
| "stop_training_after": null, | |
| "stop_training_after_unit": "NEVER", | |
| "learning_rate": null, | |
| "weight_dtype": "NONE" | |
| }, | |
| "decoder_text_encoder": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": true, | |
| "stop_training_after": null, | |
| "stop_training_after_unit": "NEVER", | |
| "learning_rate": null, | |
| "weight_dtype": "NONE" | |
| }, | |
| "decoder_vqgan": { | |
| "__version": 0, | |
| "model_name": "", | |
| "train": true, | |
| "stop_training_after": null, | |
| "stop_training_after_unit": "NEVER", | |
| "learning_rate": null, | |
| "weight_dtype": "NONE" | |
| }, | |
| "masked_training": false, | |
| "unmasked_probability": 0.1, | |
| "unmasked_weight": 0.1, | |
| "normalize_masked_area_loss": false, | |
| "embedding_learning_rate": null, | |
| "preserve_embedding_norm": false, | |
| "embedding": { | |
| "__version": 0, | |
| "uuid": "be454e9c-068a-4d34-b0e4-10cf54461eee", | |
| "model_name": "", | |
| "placeholder": "<embedding>", | |
| "train": true, | |
| "stop_training_after": null, | |
| "stop_training_after_unit": "NEVER", | |
| "token_count": 1, | |
| "initial_embedding_text": "*" | |
| }, | |
| "additional_embeddings": [], | |
| "embedding_weight_dtype": "FLOAT_32", | |
| "lora_model_name": "", | |
| "lora_rank": 16, | |
| "lora_alpha": 1.0, | |
| "lora_weight_dtype": "FLOAT_32", | |
| "optimizer": { | |
| "__version": 0, | |
| "optimizer": "ADAFACTOR", | |
| "adam_w_mode": false, | |
| "alpha": null, | |
| "amsgrad": false, | |
| "beta1": null, | |
| "beta2": null, | |
| "beta3": null, | |
| "bias_correction": false, | |
| "block_wise": false, | |
| "capturable": false, | |
| "centered": false, | |
| "clip_threshold": 1.0, | |
| "d0": null, | |
| "d_coef": null, | |
| "dampening": null, | |
| "decay_rate": -0.8, | |
| "decouple": false, | |
| "differentiable": false, | |
| "eps": 1e-30, | |
| "eps2": 0.001, | |
| "foreach": false, | |
| "fsdp_in_use": false, | |
| "fused": false, | |
| "fused_back_pass": false, | |
| "growth_rate": null, | |
| "initial_accumulator_value": null, | |
| "is_paged": false, | |
| "log_every": null, | |
| "lr_decay": null, | |
| "max_unorm": null, | |
| "maximize": false, | |
| "min_8bit_size": null, | |
| "momentum": null, | |
| "nesterov": false, | |
| "no_prox": false, | |
| "optim_bits": null, | |
| "percentile_clipping": null, | |
| "relative_step": false, | |
| "safeguard_warmup": false, | |
| "scale_parameter": false, | |
| "stochastic_rounding": false, | |
| "use_bias_correction": false, | |
| "use_triton": false, | |
| "warmup_init": false, | |
| "weight_decay": 0.01 | |
| }, | |
| "optimizer_defaults": { | |
| "ADAFACTOR": { | |
| "__version": 0, | |
| "optimizer": "ADAFACTOR", | |
| "adam_w_mode": false, | |
| "alpha": null, | |
| "amsgrad": false, | |
| "beta1": null, | |
| "beta2": null, | |
| "beta3": null, | |
| "bias_correction": false, | |
| "block_wise": false, | |
| "capturable": false, | |
| "centered": false, | |
| "clip_threshold": 1.0, | |
| "d0": null, | |
| "d_coef": null, | |
| "dampening": null, | |
| "decay_rate": -0.8, | |
| "decouple": false, | |
| "differentiable": false, | |
| "eps": 1e-30, | |
| "eps2": 0.001, | |
| "foreach": false, | |
| "fsdp_in_use": false, | |
| "fused": false, | |
| "fused_back_pass": false, | |
| "growth_rate": null, | |
| "initial_accumulator_value": null, | |
| "is_paged": false, | |
| "log_every": null, | |
| "lr_decay": null, | |
| "max_unorm": null, | |
| "maximize": false, | |
| "min_8bit_size": null, | |
| "momentum": null, | |
| "nesterov": false, | |
| "no_prox": false, | |
| "optim_bits": null, | |
| "percentile_clipping": null, | |
| "relative_step": false, | |
| "safeguard_warmup": false, | |
| "scale_parameter": false, | |
| "stochastic_rounding": false, | |
| "use_bias_correction": false, | |
| "use_triton": false, | |
| "warmup_init": false, | |
| "weight_decay": 0.01 | |
| } | |
| }, | |
| "sample_definition_file_name": "training_samples/samples.json", | |
| "samples": [], | |
| "sample_after": 10, | |
| "sample_after_unit": "MINUTE", | |
| "sample_image_format": "JPG", | |
| "samples_to_tensorboard": false, | |
| "non_ema_sampling": false, | |
| "backup_after": 30, | |
| "backup_after_unit": "NEVER", | |
| "rolling_backup": false, | |
| "rolling_backup_count": 3, | |
| "backup_before_save": false, | |
| "save_after": 50, | |
| "save_after_unit": "EPOCH", | |
| "save_filename_prefix": "brkcat" | |
| } |