From d8b066642965f738c48594a944c3f63e607b0f70 Mon Sep 17 00:00:00 2001 From: xaedes Date: Fri, 19 May 2023 18:29:47 +0200 Subject: [PATCH] initialize rng with srand --- examples/baby-llama/baby-llama-text.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/examples/baby-llama/baby-llama-text.cpp b/examples/baby-llama/baby-llama-text.cpp index 7517e2037..aa8c3ace4 100644 --- a/examples/baby-llama/baby-llama-text.cpp +++ b/examples/baby-llama/baby-llama-text.cpp @@ -3,10 +3,12 @@ #include #include #include -#include #include #include #include +#include +#include +#include #include #include @@ -1485,6 +1487,8 @@ int main(int argc, char ** argv) { //return 1; } + srand(time(NULL)); + const char * fn_model = (argc >= 2) ? argv[1] : default_argv[1]; const char * fn_train = (argc >= 3) ? argv[2] : default_argv[2]; const char * fn_chkpt = (argc >= 4) ? argv[3] : default_argv[3];