Upload OneTrainer-XLsd32-phase1-LaionPlusWD.json with huggingface_hub

OneTrainer-XLsd32-phase1-LaionPlusWD.json

@@ -0,0 +1,370 @@
+{
+    "__version": 6,
+    "training_method": "FINE_TUNE",
+    "model_type": "STABLE_DIFFUSION_15",
+    "debug_mode": false,
+    "debug_dir": "debug",
+    "workspace_dir": "workspace/run",
+    "cache_dir": "workspace-cache/run",
+    "tensorboard": true,
+    "tensorboard_expose": true,
+    "tensorboard_port": 6006,
+    "validation": false,
+    "validate_after": 1,
+    "validate_after_unit": "EPOCH",
+    "continue_last_backup": false,
+    "include_train_config": "ALL",
+    "base_model_name": "/data/OneTrainer32/models/XLsd32-base-v0.safetensors",
+    "weight_dtype": "FLOAT_32",
+    "output_dtype": "FLOAT_32",
+    "output_model_format": "SAFETENSORS",
+    "output_model_destination": "models/XLsd32-dlionb16a8-phase1-LAION.sft",
+    "gradient_checkpointing": "ON",
+    "enable_async_offloading": true,
+    "enable_activation_offloading": true,
+    "layer_offload_fraction": 0.0,
+    "force_circular_padding": false,
+    "concept_file_name": "training_concepts/XLsd32.json",
+    "concepts": null,
+    "aspect_ratio_bucketing": true,
+    "latent_caching": true,
+    "clear_cache_before_training": false,
+    "learning_rate_scheduler": "CONSTANT",
+    "custom_learning_rate_scheduler": null,
+    "scheduler_params": [],
+    "learning_rate": 1.0,
+    "learning_rate_warmup_steps": 0.0,
+    "learning_rate_cycles": 1.0,
+    "learning_rate_min_factor": 0.0,
+    "epochs": 100,
+    "batch_size": 16,
+    "gradient_accumulation_steps": 8,
+    "ema": "OFF",
+    "ema_decay": 0.999,
+    "ema_update_step_interval": 5,
+    "dataloader_threads": 8,
+    "train_device": "cuda",
+    "temp_device": "cpu",
+    "train_dtype": "FLOAT_32",
+    "fallback_train_dtype": "FLOAT_32",
+    "enable_autocast_cache": true,
+    "only_cache": false,
+    "resolution": "512",
+    "frames": "25",
+    "attention_mechanism": "SDP",
+    "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,
+    "log_cosh_strength": 0.0,
+    "vb_loss_strength": 1.0,
+    "loss_weight_fn": "DEBIASED_ESTIMATION",
+    "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,
+    "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": "UNIFORM",
+    "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
+    },
+    "prior": {
+        "__version": 0,
+        "model_name": "",
+        "include": true,
+        "train": true,
+        "stop_training_after": 0,
+        "stop_training_after_unit": "NEVER",
+        "learning_rate": null,
+        "weight_dtype": "NONE",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "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_32",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "text_encoder_layer_skip": 0,
+    "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": "NONE",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "text_encoder_2_layer_skip": 0,
+    "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": "NONE",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "text_encoder_3_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": "FLOAT_32",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "effnet_encoder": {
+        "__version": 0,
+        "model_name": "",
+        "include": true,
+        "train": true,
+        "stop_training_after": null,
+        "stop_training_after_unit": "NEVER",
+        "learning_rate": null,
+        "weight_dtype": "NONE",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "decoder": {
+        "__version": 0,
+        "model_name": "",
+        "include": true,
+        "train": true,
+        "stop_training_after": null,
+        "stop_training_after_unit": "NEVER",
+        "learning_rate": null,
+        "weight_dtype": "NONE",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "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": "NONE",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "decoder_vqgan": {
+        "__version": 0,
+        "model_name": "",
+        "include": true,
+        "train": true,
+        "stop_training_after": null,
+        "stop_training_after_unit": "NEVER",
+        "learning_rate": null,
+        "weight_dtype": "NONE",
+        "dropout_probability": 0.0,
+        "train_embedding": true,
+        "attention_mask": false,
+        "guidance_scale": 1.0
+    },
+    "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": "8ea9e884-0a1d-4aa5-9f2f-1ab6ba24ee12",
+        "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",
+    "cloud": {
+        "__version": 0,
+        "enabled": false,
+        "type": "RUNPOD",
+        "file_sync": "NATIVE_SCP",
+        "create": true,
+        "name": "OneTrainer",
+        "tensorboard_tunnel": true,
+        "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": true,
+        "update_onetrainer": true,
+        "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": 16,
+    "lora_alpha": 1.0,
+    "lora_decompose": false,
+    "lora_decompose_norm_epsilon": true,
+    "lora_weight_dtype": "FLOAT_32",
+    "lora_layers": "",
+    "lora_layer_preset": null,
+    "bundle_additional_embeddings": true,
+    "optimizer": {
+        "__version": 0,
+        "optimizer": "DADAPT_LION",
+        "adam_w_mode": false,
+        "alpha": null,
+        "amsgrad": false,
+        "beta1": 0.9,
+        "beta2": 0.999,
+        "beta3": null,
+        "bias_correction": false,
+        "block_wise": false,
+        "capturable": false,
+        "centered": false,
+        "clip_threshold": null,
+        "d0": 1e-06,
+        "d_coef": null,
+        "dampening": null,
+        "decay_rate": null,
+        "decouple": false,
+        "differentiable": false,
+        "eps": null,
+        "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": 0,
+        "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,
+        "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
+    },
+    "optimizer_defaults": {},
+    "sample_definition_file_name": "training_samples/samples.json",
+    "samples": null,
+    "sample_after": 2000,
+    "sample_after_unit": "STEP",
+    "sample_skip_first": 0,
+    "sample_image_format": "PNG",
+    "sample_video_format": "MP4",
+    "sample_audio_format": "MP3",
+    "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_every": 1,
+    "save_every_unit": "EPOCH",
+    "save_skip_first": 0,
+    "save_filename_prefix": "XLsd32-dlionb16a8-phase1-LAION-"
+}