Merge pull request #148 from damian0815/fix_default_sd21_clipskip

fix clip_skip default in trainSD21.json and update other params
This commit is contained in:
Victor Hall 2023-04-19 13:32:11 -04:00 committed by GitHub
commit 79886981a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions

View File

@ -3,7 +3,7 @@
"batch_size": 6, "batch_size": 6,
"ckpt_every_n_minutes": null, "ckpt_every_n_minutes": null,
"clip_grad_norm": null, "clip_grad_norm": null,
"clip_skip": 2, "clip_skip": 0,
"cond_dropout": 0.04, "cond_dropout": 0.04,
"data_root": "X:\\my_project_data\\project_abc", "data_root": "X:\\my_project_data\\project_abc",
"disable_textenc_training": false, "disable_textenc_training": false,
@ -21,20 +21,23 @@
"lr_warmup_steps": null, "lr_warmup_steps": null,
"max_epochs": 30, "max_epochs": 30,
"notebook": false, "notebook": false,
"optimizer_config": "optimizer.json",
"project_name": "project_abc_sd21", "project_name": "project_abc_sd21",
"resolution": 768, "resolution": 768,
"resume_ckpt": "v2-1_768-nonema-pruned", "resume_ckpt": "v2-1_768-nonema-pruned",
"sample_prompts": "sample_prompts.txt", "sample_prompts": "sample_prompts.txt",
"sample_steps": 300, "sample_steps": 300,
"save_ckpt_dir": null, "save_ckpt_dir": null,
"save_ckpts_from_n_epochs": 0,
"save_every_n_epochs": 20, "save_every_n_epochs": 20,
"save_optimizer": false, "save_optimizer": false,
"scale_lr": false, "scale_lr": false,
"seed": 555, "seed": 555,
"shuffle_tags": false, "shuffle_tags": false,
"useadam8bit": true, "validation_config": "validation_default.json",
"wandb": false, "wandb": false,
"write_schedule": false, "write_schedule": false,
"rated_dataset": false, "rated_dataset": false,
"rated_dataset_target_dropout_rate": 50 "rated_dataset_target_dropout_percent": 50,
"zero_frequency_noise_ratio": 0.02
} }