Add new test cases

test/config/dreambooth-AdamW.json

@@ -0,0 +1,73 @@
+{
+  "adaptive_noise_scale": 0,
+  "additional_parameters": "",
+  "bucket_no_upscale": true,
+  "bucket_reso_steps": 64,
+  "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,
+  "enable_bucket": true,
+  "epoch": 1,
+  "flip_aug": false,
+  "full_fp16": false,
+  "gradient_accumulation_steps": 1.0,
+  "gradient_checkpointing": false,
+  "keep_tokens": "0",
+  "learning_rate": 5e-05,
+  "logging_dir": "./test/logs",
+  "lr_scheduler": "constant",
+  "lr_warmup": 0,
+  "max_data_loader_n_workers": "0",
+  "max_resolution": "512,512",
+  "max_token_length": "75",
+  "max_train_epochs": "",
+  "mem_eff_attn": false,
+  "min_snr_gamma": 0,
+  "mixed_precision": "bf16",
+  "model_list": "runwayml/stable-diffusion-v1-5",
+  "multires_noise_discount": 0,
+  "multires_noise_iterations": 0,
+  "no_token_padding": false,
+  "noise_offset": "0.05",
+  "noise_offset_type": "Original",
+  "num_cpu_threads_per_process": 2,
+  "optimizer": "AdamW",
+  "optimizer_args": "",
+  "output_dir": "./test/output",
+  "output_name": "db-AdamW",
+  "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": 25,
+  "sample_prompts": "a painting of 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,
+  "seed": "1234",
+  "shuffle_caption": false,
+  "stop_text_encoder_training": 0,
+  "train_batch_size": 4,
+  "train_data_dir": "./test/img",
+  "use_wandb": false,
+  "v2": false,
+  "v_parameterization": false,
+  "vae": "",
+  "vae_batch_size": 0,
+  "wandb_api_key": "",
+  "weighted_captions": false,
+  "xformers": true
+}

test/config/dreambooth-AdamW8bit.json

@@ -17,7 +17,7 @@
   "gradient_accumulation_steps": 1.0,
   "gradient_checkpointing": false,
   "keep_tokens": "0",
-  "learning_rate": 1.0,
+  "learning_rate": 5e-05,
   "logging_dir": "./test/logs",
   "lr_scheduler": "constant",
   "lr_warmup": 0,
@@ -38,7 +38,7 @@
   "optimizer": "AdamW8bit",
   "optimizer_args": "",
   "output_dir": "./test/output",
-  "output_name": "db-8bit",
+  "output_name": "db-AdamW8bit",
   "persistent_data_loader_workers": false,
   "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5",
   "prior_loss_weight": 1.0,

test/config/dreambooth-DAdaptAdam.json

@@ -0,0 +1,73 @@
+{
+  "adaptive_noise_scale": 0,
+  "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,
+  "caption_extension": "",
+  "clip_skip": 2,
+  "color_aug": false,
+  "enable_bucket": true,
+  "epoch": 1,
+  "flip_aug": false,
+  "full_fp16": false,
+  "gradient_accumulation_steps": 1.0,
+  "gradient_checkpointing": false,
+  "keep_tokens": "0",
+  "learning_rate": 1.0,
+  "logging_dir": "./test/logs",
+  "lr_scheduler": "cosine",
+  "lr_warmup": 0,
+  "max_data_loader_n_workers": "0",
+  "max_resolution": "512,512",
+  "max_token_length": "75",
+  "max_train_epochs": "",
+  "mem_eff_attn": false,
+  "min_snr_gamma": 0,
+  "mixed_precision": "bf16",
+  "model_list": "runwayml/stable-diffusion-v1-5",
+  "multires_noise_discount": 0.2,
+  "multires_noise_iterations": 8,
+  "no_token_padding": false,
+  "noise_offset": "0.05",
+  "noise_offset_type": "Multires",
+  "num_cpu_threads_per_process": 2,
+  "optimizer": "DAdaptAdam",
+  "optimizer_args": "decouple=True weight_decay=0.6 betas=0.9,0.99 use_bias_correction=True",
+  "output_dir": "./test/output",
+  "output_name": "db-DAdaptAdam",
+  "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": 25,
+  "sample_prompts": "a painting of 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,
+  "seed": "1234",
+  "shuffle_caption": false,
+  "stop_text_encoder_training": 0,
+  "train_batch_size": 1,
+  "train_data_dir": "./test/img",
+  "use_wandb": false,
+  "v2": false,
+  "v_parameterization": false,
+  "vae": "",
+  "vae_batch_size": 0,
+  "wandb_api_key": "",
+  "weighted_captions": false,
+  "xformers": true
+}

test/config/dreambooth-Prodigy.json

@@ -0,0 +1,73 @@
+{
+  "adaptive_noise_scale": 0,
+  "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,
+  "caption_extension": "",
+  "clip_skip": 2,
+  "color_aug": false,
+  "enable_bucket": true,
+  "epoch": 1,
+  "flip_aug": false,
+  "full_fp16": false,
+  "gradient_accumulation_steps": 1.0,
+  "gradient_checkpointing": false,
+  "keep_tokens": "0",
+  "learning_rate": 1.0,
+  "logging_dir": "./test/logs",
+  "lr_scheduler": "cosine",
+  "lr_warmup": 0,
+  "max_data_loader_n_workers": "0",
+  "max_resolution": "512,512",
+  "max_token_length": "75",
+  "max_train_epochs": "",
+  "mem_eff_attn": false,
+  "min_snr_gamma": 0,
+  "mixed_precision": "bf16",
+  "model_list": "runwayml/stable-diffusion-v1-5",
+  "multires_noise_discount": 0.2,
+  "multires_noise_iterations": 8,
+  "no_token_padding": false,
+  "noise_offset": "0.05",
+  "noise_offset_type": "Multires",
+  "num_cpu_threads_per_process": 2,
+  "optimizer": "Prodigy",
+  "optimizer_args": "decouple=True weight_decay=0.6 betas=0.9,0.99 use_bias_correction=True",
+  "output_dir": "./test/output",
+  "output_name": "db-Prodigy",
+  "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": 25,
+  "sample_prompts": "a painting of 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,
+  "seed": "1234",
+  "shuffle_caption": false,
+  "stop_text_encoder_training": 0,
+  "train_batch_size": 1,
+  "train_data_dir": "./test/img",
+  "use_wandb": false,
+  "v2": false,
+  "v_parameterization": false,
+  "vae": "",
+  "vae_batch_size": 0,
+  "wandb_api_key": "",
+  "weighted_captions": false,
+  "xformers": true
+}

test/config/locon-AdamW.json

@@ -0,0 +1,101 @@
+{
+  "LoRA_type": "Kohya LoCon",
+  "adaptive_noise_scale": 0,
+  "additional_parameters": "",
+  "block_alphas": "",
+  "block_dims": "",
+  "block_lr_zero_threshold": "",
+  "bucket_no_upscale": true,
+  "bucket_reso_steps": 64,
+  "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,
+  "conv_alpha": 8,
+  "conv_alphas": "",
+  "conv_dim": 16,
+  "conv_dims": "",
+  "decompose_both": false,
+  "dim_from_weights": false,
+  "down_lr_weight": "",
+  "enable_bucket": true,
+  "epoch": 1,
+  "factor": -1,
+  "flip_aug": false,
+  "full_fp16": false,
+  "gradient_accumulation_steps": 1,
+  "gradient_checkpointing": false,
+  "keep_tokens": "0",
+  "learning_rate": 0.0005,
+  "logging_dir": "./test/logs",
+  "lora_network_weights": "",
+  "lr_scheduler": "constant",
+  "lr_scheduler_num_cycles": "",
+  "lr_scheduler_power": "",
+  "lr_warmup": 0,
+  "max_data_loader_n_workers": "0",
+  "max_resolution": "512,512",
+  "max_token_length": "75",
+  "max_train_epochs": "",
+  "mem_eff_attn": false,
+  "mid_lr_weight": "",
+  "min_snr_gamma": 0,
+  "mixed_precision": "bf16",
+  "model_list": "runwayml/stable-diffusion-v1-5",
+  "module_dropout": 0.1,
+  "multires_noise_discount": 0,
+  "multires_noise_iterations": 0,
+  "network_alpha": 8,
+  "network_dim": 16,
+  "network_dropout": 0.1,
+  "no_token_padding": false,
+  "noise_offset": "0.05",
+  "noise_offset_type": "Original",
+  "num_cpu_threads_per_process": 2,
+  "optimizer": "AdamW",
+  "optimizer_args": "",
+  "output_dir": "./test/output",
+  "output_name": "locon-AdamW",
+  "persistent_data_loader_workers": false,
+  "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "prior_loss_weight": 1.0,
+  "random_crop": false,
+  "rank_dropout": 0.1,
+  "reg_data_dir": "",
+  "resume": "",
+  "sample_every_n_epochs": 0,
+  "sample_every_n_steps": 25,
+  "sample_prompts": "a painting of 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,
+  "scale_weight_norms": 1,
+  "seed": "1234",
+  "shuffle_caption": false,
+  "stop_text_encoder_training": 0,
+  "text_encoder_lr": 0.0001,
+  "train_batch_size": 4,
+  "train_data_dir": "./test/img",
+  "train_on_input": false,
+  "training_comment": "",
+  "unet_lr": 0.0001,
+  "unit": 1,
+  "up_lr_weight": "",
+  "use_cp": false,
+  "use_wandb": false,
+  "v2": false,
+  "v_parameterization": false,
+  "vae_batch_size": 0,
+  "wandb_api_key": "",
+  "weighted_captions": false,
+  "xformers": true
+}