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

feat: set default rng to cuda

Browse Source
This commit is contained in:
leejet 2023-09-04 21:46:54 +08:00
parent 34a118d407
commit b85b236b13
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/main.cpp
View File

@ -86,7 +86,7 @@ struct Option {
SampleMethod sample_method = EULAR_A; SampleMethod sample_method = EULAR_A;
int sample_steps = 20; int sample_steps = 20;
float strength = 0.75f; float strength = 0.75f;
RNGType rng_type = STD_DEFAULT_RNG; RNGType rng_type = CUDA_RNG;
int64_t seed = 42; int64_t seed = 42;
bool verbose = false; bool verbose = false;
@ -130,7 +130,7 @@ void print_usage(int argc, const char* argv[]) {
printf(" -W, --width W image width, in pixel space (default: 512)\n"); printf(" -W, --width W image width, in pixel space (default: 512)\n");
printf(" --sample-method SAMPLE_METHOD sample method (default: \"eular a\")\n"); printf(" --sample-method SAMPLE_METHOD sample method (default: \"eular a\")\n");
printf(" --steps STEPS number of sample steps (default: 20)\n"); printf(" --steps STEPS number of sample steps (default: 20)\n");
printf(" --rng {std_default, cuda} RNG (default: std_default)\n"); 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(" -s SEED, --seed SEED RNG seed (default: 42, use random seed for < 0)\n");
printf(" -v, --verbose print extra info\n"); printf(" -v, --verbose print extra info\n");
} }