Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- {
- "__version": 11,
- "training_method": "LORA",
- "model_type": "KREA_2",
- "debug_mode": false,
- "debug_dir": "debug",
- "workspace_dir": "workspace/run",
- "cache_dir": "workspace-cache/run",
- "tensorboard": false,
- "tensorboard_expose": false,
- "tensorboard_always_on": false,
- "tensorboard_port": 6006,
- "validation": false,
- "validate_after": 1,
- "validate_after_unit": "EPOCH",
- "continue_last_backup": false,
- "prevent_overwrites": false,
- "include_train_config": "NONE",
- "multi_gpu": false,
- "device_indexes": "",
- "gradient_reduce_precision": "FLOAT_32_STOCHASTIC",
- "fused_gradient_reduce": true,
- "async_gradient_reduce": true,
- "async_gradient_reduce_buffer": 100,
- "base_model_name": "krea/Krea-2-Raw",
- "output_dtype": "BFLOAT_16",
- "output_model_format": "DIFFUSERS_LORA",
- "output_model_destination": "LORA PATH HERE",
- "async_offloading": true,
- "force_circular_padding": false,
- "compile": true,
- "concept_file_name": "training_concepts/concepts.json",
- "concepts": null,
- "aspect_ratio_bucketing": true,
- "latent_caching": true,
- "clear_cache_before_training": true,
- "learning_rate_scheduler": "COSINE",
- "custom_learning_rate_scheduler": null,
- "scheduler_params": [],
- "learning_rate": 0.0001,
- "learning_rate_warmup_steps": 0.1,
- "learning_rate_cycles": 1.0,
- "learning_rate_min_factor": 0.0,
- "epochs": 45,
- "batch_size": 1,
- "gradient_accumulation_steps": 2,
- "ema": "OFF",
- "ema_decay": 0.999,
- "ema_update_step_interval": 5,
- "dataloader_threads": 2,
- "train_device": "cuda",
- "temp_device": "cpu",
- "train_dtype": "BFLOAT_16",
- "fallback_train_dtype": "BFLOAT_16",
- "enable_autocast_cache": true,
- "only_cache": false,
- "resolution": "512",
- "frames": "25",
- "attention_mechanism": "CUDNN",
- "mse_strength": 1.0,
- "mae_strength": 0.0,
- "log_cosh_strength": 0.0,
- "huber_strength": 0.0,
- "huber_delta": 1.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",
- "clip_grad_norm": 1.0,
- "offset_noise_weight": 0.0,
- "generalized_offset_noise": false,
- "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,
- "timestep_distribution": "LOGIT_NORMAL",
- "noising_weight": 0.0,
- "noising_bias": 0.0,
- "timestep_shift": 1.0,
- "dynamic_timestep_shifting": false,
- "unet": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": 0,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "prior": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": 0,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "transformer": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": 0,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "INT_W8A8",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "unconditional_transformer": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": false,
- "stop_training_after": null,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": false,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "quantization": {
- "__version": 0,
- "layer_filter": "attn,ff",
- "layer_filter_preset": "attn-mlp",
- "layer_filter_regex": false,
- "svd_dtype": "NONE",
- "svd_rank": 16,
- "cache_dir": "workspace-cache/run/quantization"
- },
- "text_encoder": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": false,
- "stop_training_after": 30,
- "stop_training_after_unit": "EPOCH",
- "learning_rate": null,
- "weight_dtype": "FLOAT_8",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "text_encoder_layer_skip": 0,
- "text_encoder_sequence_length": 512,
- "text_encoder_2": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": 30,
- "stop_training_after_unit": "EPOCH",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "text_encoder_2_layer_skip": 0,
- "text_encoder_2_sequence_length": 77,
- "text_encoder_3": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": 30,
- "stop_training_after_unit": "EPOCH",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "text_encoder_3_layer_skip": 0,
- "text_encoder_4": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": 30,
- "stop_training_after_unit": "EPOCH",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "text_encoder_4_layer_skip": 0,
- "vae": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": null,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "BFLOAT_16",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "effnet_encoder": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": null,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "decoder": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": null,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "decoder_text_encoder": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": null,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "decoder_vqgan": {
- "__version": 0,
- "model_name": "",
- "include": true,
- "train": true,
- "stop_training_after": null,
- "stop_training_after_unit": "NEVER",
- "learning_rate": null,
- "weight_dtype": "FLOAT_32",
- "dropout_probability": 0.0,
- "train_embedding": true,
- "attention_mask": false,
- "guidance_scale": 1.0,
- "gradient_checkpointing": true,
- "offload_fraction": 0.0,
- "activation_offloading": false
- },
- "masked_training": false,
- "unmasked_probability": 0.1,
- "unmasked_weight": 0.1,
- "normalize_masked_area_loss": false,
- "masked_prior_preservation_weight": 0.0,
- "custom_conditioning_image": false,
- "layer_filter": "attn,ff",
- "layer_filter_preset": "attn-mlp",
- "layer_filter_regex": false,
- "embedding_learning_rate": null,
- "preserve_embedding_norm": false,
- "embedding": {
- "__version": 0,
- "uuid": "",
- "model_name": "",
- "placeholder": "<embedding>",
- "train": true,
- "stop_training_after": null,
- "stop_training_after_unit": "NEVER",
- "token_count": 1,
- "initial_embedding_text": "*",
- "is_output_embedding": false
- },
- "additional_embeddings": [],
- "embedding_weight_dtype": "FLOAT_32",
- "cloud": {
- "__version": 0,
- "enabled": false,
- "type": "RUNPOD",
- "file_sync": "NATIVE_SCP",
- "create": false,
- "name": "OneTrainer",
- "tensorboard_tunnel": false,
- "sub_type": "",
- "gpu_type": "",
- "volume_size": 100,
- "min_download": 0,
- "remote_dir": "/workspace",
- "huggingface_cache_dir": "/workspace/huggingface_cache",
- "onetrainer_dir": "/workspace/OneTrainer",
- "install_cmd": "git clone https://github.com/Nerogar/OneTrainer",
- "install_onetrainer": false,
- "update_onetrainer": false,
- "detach_trainer": false,
- "run_id": "job1",
- "download_samples": true,
- "download_output_model": true,
- "download_saves": true,
- "download_backups": false,
- "download_tensorboard": false,
- "delete_workspace": false,
- "on_finish": "NONE",
- "on_error": "NONE",
- "on_detached_finish": "NONE",
- "on_detached_error": "NONE"
- },
- "peft_type": "LORA",
- "lora_model_name": "",
- "lora_rank": 32,
- "lora_alpha": 32.0,
- "lora_decompose": false,
- "lora_decompose_norm_epsilon": true,
- "lora_decompose_output_axis": false,
- "lora_weight_dtype": "FLOAT_32",
- "bundle_additional_embeddings": true,
- "oft_block_size": 32,
- "oft_block_share": false,
- "oft_scaled": false,
- "lokr_dim": 16,
- "lokr_decompose_both": false,
- "lokr_decompose_factor": -1,
- "lokr_use_tucker": false,
- "lokr_weight_decompose": false,
- "lokr_dora_on_output": true,
- "lokr_full_matrix": false,
- "lokr_vec_trick": true,
- "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,
- "initial_accumulator": null,
- "is_paged": false,
- "log_every": null,
- "lr_decay": null,
- "max_unorm": null,
- "maximize": false,
- "min_8bit_size": null,
- "quant_block_size": null,
- "momentum": null,
- "nesterov": false,
- "no_prox": false,
- "optim_bits": null,
- "percentile_clipping": null,
- "r": null,
- "relative_step": false,
- "safeguard_warmup": false,
- "scale_parameter": false,
- "stochastic_rounding": true,
- "use_bias_correction": false,
- "use_triton": false,
- "warmup_init": false,
- "weight_decay": 0.0,
- "weight_lr_power": null,
- "decoupled_decay": false,
- "fixed_decay": false,
- "rectify": false,
- "degenerated_to_sgd": false,
- "k": null,
- "xi": null,
- "n_sma_threshold": null,
- "ams_bound": false,
- "adanorm": false,
- "adam_debias": false,
- "slice_p": null,
- "cautious": false,
- "weight_decay_by_lr": true,
- "prodigy_steps": null,
- "use_speed": false,
- "split_groups": true,
- "split_groups_mean": true,
- "factored": true,
- "factored_fp32": true,
- "use_stableadamw": true,
- "use_cautious": false,
- "use_grams": false,
- "use_adopt": false,
- "d_limiter": true,
- "use_schedulefree": true,
- "use_orthograd": false,
- "nnmf_factor": false,
- "orthogonal_gradient": false,
- "use_atan2": false,
- "use_AdEMAMix": false,
- "beta3_ema": null,
- "alpha_grad": null,
- "beta1_warmup": null,
- "min_beta1": null,
- "Simplified_AdEMAMix": false,
- "kourkoutas_beta": false,
- "schedulefree_c": null,
- "ns_steps": null,
- "MuonWithAuxAdam": false,
- "muon_hidden_layers": null,
- "muon_adam_regex": false,
- "muon_adam_lr": null,
- "muon_te1_adam_lr": null,
- "muon_te2_adam_lr": null,
- "muon_adam_config": {},
- "rms_rescaling": true,
- "normuon_variant": false,
- "beta2_normuon": null,
- "low_rank_ortho": false,
- "ortho_rank": null,
- "accelerated_ns": false,
- "cautious_wd": false,
- "approx_mars": false,
- "auto_kappa_p": false,
- "compile": false
- },
- "optimizer_defaults": {
- "ADAMW_8BIT": {
- "__version": 0,
- "optimizer": "ADAMW_8BIT",
- "adam_w_mode": false,
- "alpha": null,
- "amsgrad": false,
- "beta1": 0.9,
- "beta2": 0.999,
- "beta3": null,
- "bias_correction": false,
- "block_wise": true,
- "capturable": false,
- "centered": false,
- "clip_threshold": null,
- "d0": null,
- "d_coef": null,
- "dampening": null,
- "decay_rate": null,
- "decouple": false,
- "differentiable": false,
- "eps": 1e-08,
- "eps2": null,
- "foreach": false,
- "fsdp_in_use": false,
- "fused": false,
- "fused_back_pass": false,
- "growth_rate": null,
- "initial_accumulator_value": null,
- "initial_accumulator": null,
- "is_paged": false,
- "log_every": null,
- "lr_decay": null,
- "max_unorm": null,
- "maximize": false,
- "min_8bit_size": 4096,
- "quant_block_size": null,
- "momentum": null,
- "nesterov": false,
- "no_prox": false,
- "optim_bits": 32,
- "percentile_clipping": 100,
- "r": null,
- "relative_step": false,
- "safeguard_warmup": false,
- "scale_parameter": false,
- "stochastic_rounding": true,
- "use_bias_correction": false,
- "use_triton": false,
- "warmup_init": false,
- "weight_decay": 0.01,
- "weight_lr_power": null,
- "decoupled_decay": false,
- "fixed_decay": false,
- "rectify": false,
- "degenerated_to_sgd": false,
- "k": null,
- "xi": null,
- "n_sma_threshold": null,
- "ams_bound": false,
- "adanorm": false,
- "adam_debias": false,
- "slice_p": null,
- "cautious": false,
- "weight_decay_by_lr": true,
- "prodigy_steps": null,
- "use_speed": false,
- "split_groups": true,
- "split_groups_mean": true,
- "factored": true,
- "factored_fp32": true,
- "use_stableadamw": true,
- "use_cautious": false,
- "use_grams": false,
- "use_adopt": false,
- "d_limiter": true,
- "use_schedulefree": true,
- "use_orthograd": false,
- "nnmf_factor": false,
- "orthogonal_gradient": false,
- "use_atan2": false,
- "use_AdEMAMix": false,
- "beta3_ema": null,
- "alpha_grad": null,
- "beta1_warmup": null,
- "min_beta1": null,
- "Simplified_AdEMAMix": false,
- "kourkoutas_beta": false,
- "schedulefree_c": null,
- "ns_steps": null,
- "MuonWithAuxAdam": false,
- "muon_hidden_layers": null,
- "muon_adam_regex": false,
- "muon_adam_lr": null,
- "muon_te1_adam_lr": null,
- "muon_te2_adam_lr": null,
- "muon_adam_config": {},
- "rms_rescaling": true,
- "normuon_variant": false,
- "beta2_normuon": null,
- "low_rank_ortho": false,
- "ortho_rank": null,
- "accelerated_ns": false,
- "cautious_wd": false,
- "approx_mars": false,
- "auto_kappa_p": false,
- "compile": false
- }
- },
- "sample_definition_file_name": "training_samples/samples.json",
- "samples": null,
- "sample_after": 100,
- "sample_after_unit": "EPOCH",
- "sample_skip_first": 0,
- "sample_image_format": "JPG",
- "sample_video_format": "MP4",
- "sample_audio_format": "MP3",
- "samples_to_tensorboard": false,
- "non_ema_sampling": true,
- "backup_after": 1,
- "backup_after_unit": "EPOCH",
- "rolling_backup": true,
- "rolling_backup_count": 10,
- "backup_before_save": true,
- "save_every": 1,
- "save_every_unit": "EPOCH",
- "save_skip_first": 0,
- "save_filename_prefix": ""
- }
Advertisement
Add Comment
Please, Sign In to add comment