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 <unordered_map>
 #include <vector>
 #include <cassert>
-#include <random>
 #include <cstring>
 #include <stdexcept>
 #include <cstdarg>
+#include <ctime>
+#include <random>
+#include <stdexcept>
 #include <algorithm>
 #include <string>
 
@@ -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];