From 2fd513a82670d2092ea75d0e209d06f1db27fee1 Mon Sep 17 00:00:00 2001
From: slaren <slarengh@gmail.com>
Date: Mon, 9 Sep 2024 16:06:05 +0200
Subject: [PATCH] match discrete_dist type and function return type

---
 src/llama-sampling.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llama-sampling.cpp b/src/llama-sampling.cpp
index deca2333c..6f448b80c 100644
--- a/src/llama-sampling.cpp
+++ b/src/llama-sampling.cpp
@@ -40,7 +40,7 @@ static int llama_sample_dist(llama_token_data_array * cur_p, std::mt19937 & rng)
     #pragma GCC diagnostic pop
 #endif
 
-    std::discrete_distribution<size_t> dist(probs_iterator{cur_p->data}, probs_iterator{cur_p->data + cur_p->size});
+    std::discrete_distribution<int> dist(probs_iterator{cur_p->data}, probs_iterator{cur_p->data + cur_p->size});
 
     return dist(rng);
 }