From b85b236b134de34d0cb854d2a5a14013b9581f27 Mon Sep 17 00:00:00 2001 From: leejet Date: Mon, 4 Sep 2023 21:46:54 +0800 Subject: [PATCH] feat: set default rng to cuda --- examples/main.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/main.cpp b/examples/main.cpp index fd2c9c9..6cc4e68 100644 --- a/examples/main.cpp +++ b/examples/main.cpp @@ -86,7 +86,7 @@ struct Option { SampleMethod sample_method = EULAR_A; int sample_steps = 20; float strength = 0.75f; - RNGType rng_type = STD_DEFAULT_RNG; + RNGType rng_type = CUDA_RNG; int64_t seed = 42; 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(" --sample-method SAMPLE_METHOD sample method (default: \"eular a\")\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(" -v, --verbose print extra info\n"); }