This commit is contained in:
Rokket 2026-03-23 17:51:06 -05:00 committed by GitHub
commit 6227cda9ad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions

View File

@ -501,10 +501,11 @@ task_params server_task::params_from_json_cmpl(
}
{
params.sampling.logit_bias.clear();
params.sampling.logit_bias = params_base.sampling.logit_bias;
const auto & logit_bias = data.find("logit_bias");
if (logit_bias != data.end() && logit_bias->is_array()) {
params.sampling.logit_bias.clear();
const int n_vocab = llama_vocab_n_tokens(vocab);
for (const auto & el : *logit_bias) {
// TODO: we may want to throw errors here, in case "el" is incorrect
@ -532,6 +533,7 @@ task_params server_task::params_from_json_cmpl(
}
}
} else if (logit_bias != data.end() && logit_bias->is_object()) {
params.sampling.logit_bias.clear();
const int n_vocab = llama_vocab_n_tokens(vocab);
for (const auto & el : logit_bias->items()) {
float bias;