public_git/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggml-org/whisper.cpp.git synced 2026-09-30 03:21:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

common : fix for two functions when top_k exceeds the vocabulary size. (ggml/1633)

Browse Source
This commit is contained in:
Arianna Method
2026-09-23 20:46:47 +03:00
committed by Georgi Gerganov
parent f6b039ff0f
commit 711ef84442
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/common.cpp
+2
View File
@@ -406,6 +406,7 @@ gpt_vocab::id gpt_sample_top_k_top_p(
double temp,
std::mt19937 & rng) {
int n_logits = vocab.id_to_token.size();
top_k = std::min(top_k, n_logits);
std::vector<std::pair<double, gpt_vocab::id>> logits_id;
logits_id.reserve(n_logits);
@@ -491,6 +492,7 @@ gpt_vocab::id gpt_sample_top_k_top_p_repeat(
std::mt19937 & rng) {
int n_logits = vocab.id_to_token.size();
top_k = std::min(top_k, n_logits);
const auto * plogits = logits;