From 4f546091102a418ffdc6230f872ac56e5cedb835 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Fri, 17 Mar 2023 21:46:46 +0200
Subject: Default to 4 threads (#243)

---
 utils.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils.h b/utils.h
index 5e5b40f..c1a8498 100644
--- a/utils.h
+++ b/utils.h
@@ -14,11 +14,11 @@
 
 struct gpt_params {
     int32_t seed      = -1; // RNG seed
-    int32_t n_threads;
+    int32_t n_threads = std::min(4, (int32_t) std::thread::hardware_concurrency());
     int32_t n_predict = 128; // new tokens to predict
     int32_t repeat_last_n = 64;  // last n tokens to penalize
     int32_t n_ctx = 512; //context size
-    
+
     // sampling parameters
     int32_t top_k = 40;
     float   top_p = 0.95f;
-- 
cgit v1.2.3