sd21 example json config

trainSD21.json

@@ -0,0 +1,40 @@
+{
+  "amp": true,
+  "batch_size": 6,
+  "ckpt_every_n_minutes": null,
+  "clip_grad_norm": null,
+  "clip_skip": 2,
+  "cond_dropout": 0.04,
+  "data_root": "X:\\my_project_data\\project_abc",
+  "disable_textenc_training": false,
+  "disable_xformers": false,
+  "flip_p": 0.0,
+  "gpuid": 0,
+  "gradient_checkpointing": true,
+  "grad_accum": 1,
+  "logdir": "logs",
+  "log_step": 25,
+  "lowvram": false,
+  "lr": 0.6e-06,
+  "lr_decay_steps": 0,
+  "lr_scheduler": "constant",
+  "lr_warmup_steps": null,
+  "max_epochs": 30,
+  "notebook": false,
+  "project_name": "project_abc_sd21",
+  "resolution": 768,
+  "resume_ckpt": "v2-1_768-nonema-pruned",
+  "sample_prompts": "sample_prompts.txt",
+  "sample_steps": 300,
+  "save_ckpt_dir": null,
+  "save_every_n_epochs": 20,
+  "save_optimizer": false,
+  "scale_lr": false,
+  "seed": 555,
+  "shuffle_tags": false,
+  "useadam8bit": true,
+  "wandb": false,
+  "write_schedule": false,
+  "rated_dataset": false,
+  "rated_dataset_target_dropout_rate": 50
+}