alphard/stable-diffusion.cpp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add lcm sampler support

Browse Source 
This referenced an issue discussion of the stable-diffusion-webui at
https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/13952, which
may not be too perfect.
This commit is contained in:
leejet
2023-11-17 22:50:06 +08:00
parent 3bf1665885
commit 536f3af672
5 changed files with 384 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/main.cpp
View File

@@ -80,13 +80,16 @@ const char* sample_method_str[] = {
"dpm2",
"dpm++2s_a",
"dpm++2m",
"dpm++2mv2"};
"dpm++2mv2",
"lcm",
};
// Names of the sigma schedule overrides, same order as Schedule in stable-diffusion.h
const char* schedule_str[] = {
"default",
"discrete",
"karras"};
"karras"
};
struct Option {
int n_threads = -1;
@@ -146,7 +149,7 @@ void print_usage(int argc, const char* argv[]) {
printf(" 1.0 corresponds to full destruction of information in init image\n");
printf(" -H, --height H image height, in pixel space (default: 512)\n");
printf(" -W, --width W image width, in pixel space (default: 512)\n");
printf(" --sampling-method {euler, euler_a, heun, dpm2, dpm++2s_a, dpm++2m, dpm++2mv2}\n");
printf(" --sampling-method {euler, euler_a, heun, dpm2, dpm++2s_a, dpm++2m, dpm++2mv2, lcm}\n");
printf(" sampling method (default: \"euler_a\")\n");
printf(" --steps STEPS number of sample steps (default: 20)\n");
printf(" --rng {std_default, cuda} RNG (default: cuda)\n");