Merge pull request #3076 from Beinsezii/vllm_n_fix

Remove `best_of` for VLLM
This commit is contained in:
Cohee
2024-11-16 15:51:25 +02:00
committed by GitHub

View File

@@ -1209,7 +1209,6 @@ export function getTextGenGenerationData(finalPrompt, maxTokens, isImpersonate,
};
const vllmParams = {
'n': canMultiSwipe ? settings.n : 1,
'best_of': canMultiSwipe ? settings.n : 1,
'ignore_eos': settings.ignore_eos_token,
'spaces_between_special_tokens': settings.spaces_between_special_tokens,
'seed': settings.seed >= 0 ? settings.seed : undefined,