Skip to content

Commit b85b236

Browse files
committed
feat: set default rng to cuda
1 parent 34a118d commit b85b236

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/main.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ struct Option {
8686
SampleMethod sample_method = EULAR_A;
8787
int sample_steps = 20;
8888
float strength = 0.75f;
89-
RNGType rng_type = STD_DEFAULT_RNG;
89+
RNGType rng_type = CUDA_RNG;
9090
int64_t seed = 42;
9191
bool verbose = false;
9292

@@ -130,7 +130,7 @@ void print_usage(int argc, const char* argv[]) {
130130
printf(" -W, --width W image width, in pixel space (default: 512)\n");
131131
printf(" --sample-method SAMPLE_METHOD sample method (default: \"eular a\")\n");
132132
printf(" --steps STEPS number of sample steps (default: 20)\n");
133-
printf(" --rng {std_default, cuda} RNG (default: std_default)\n");
133+
printf(" --rng {std_default, cuda} RNG (default: cuda)\n");
134134
printf(" -s SEED, --seed SEED RNG seed (default: 42, use random seed for < 0)\n");
135135
printf(" -v, --verbose print extra info\n");
136136
}

0 commit comments

Comments
 (0)