feat: add GITS scheduler (#343)

This commit is contained in:
Daniele
2024-08-27 16:02:17 +00:00
committed by GitHub
parent 353ee93e2d
commit 2d4a2f7982
6 changed files with 474 additions and 83 deletions

View File

@@ -45,6 +45,7 @@ const char* schedule_str[] = {
"discrete",
"karras",
"ays",
"gits",
};
const char* modes_str[] = {
@@ -200,7 +201,7 @@ void print_usage(int argc, const char* argv[]) {
printf(" --rng {std_default, cuda} RNG (default: cuda)\n");
printf(" -s SEED, --seed SEED RNG seed (default: 42, use random seed for < 0)\n");
printf(" -b, --batch-count COUNT number of images to generate.\n");
printf(" --schedule {discrete, karras, ays} Denoiser sigma schedule (default: discrete)\n");
printf(" --schedule {discrete, karras, ays, gits} Denoiser sigma schedule (default: discrete)\n");
printf(" --clip-skip N ignore last layers of CLIP network; 1 ignores none, 2 ignores one layer (default: -1)\n");
printf(" <= 0 represents unspecified, will be 1 for SD1.x, 2 for SD2.x\n");
printf(" --vae-tiling process vae in tiles to reduce memory usage\n");