tests : init prob correctly

ggerganov · ggerganov · commit baf3d3763d78 · 2024-10-17T18:23:02.000+03:00
diff --git a/tests/test-sampling.cpp b/tests/test-sampling.cpp
@@ -33,7 +33,7 @@ struct sampler_tester {
         cur.reserve(probs.size());
         for (llama_token token_id = 0; token_id < (llama_token)probs.size(); token_id++) {
             const float logit = logf(probs[token_id]);
-            cur.emplace_back(llama_token_data{token_id, logit, 0.0f});
+            cur.emplace_back(llama_token_data{token_id, logit, probs[token_id]});
         }
 
         cur_p = llama_token_data_array { cur.data(), cur.size(), -1, false };

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@ struct sampler_tester {`
`33`	`33`	`cur.reserve(probs.size());`
`34`	`34`	`for (llama_token token_id = 0; token_id < (llama_token)probs.size(); token_id++) {`
`35`	`35`	`const float logit = logf(probs[token_id]);`
`36`		`- cur.emplace_back(llama_token_data{token_id, logit, 0.0f});`
	`36`	`+ cur.emplace_back(llama_token_data{token_id, logit, probs[token_id]});`
`37`	`37`	`}`
`38`	`38`
`39`	`39`	`cur_p = llama_token_data_array { cur.data(), cur.size(), -1, false };`