Skip to content

Commit 8e68a18

Browse files
committed
feat(vllm): Add default temperature and top_p to generation params
1 parent 62c9a6a commit 8e68a18

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

lmms_eval/models/vllm.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -303,9 +303,15 @@ def generate_until(self, requests) -> List[str]:
303303
contexts, gen_kwargs, doc_to_visual, doc_id, task, split = batch_requests[idx].arguments
304304
if "max_new_tokens" not in gen_kwargs:
305305
gen_kwargs["max_new_tokens"] = 1024
306+
if "temperature" not in gen_kwargs:
307+
gen_kwargs["temperature"] = 0
308+
if "top_p" not in gen_kwargs:
309+
gen_kwargs["top_p"] = 0.95
306310

307311
params = {
308312
"max_tokens": gen_kwargs["max_new_tokens"],
313+
"temperature": gen_kwargs["temperature"],
314+
"top_p": gen_kwargs["top_p"],
309315
}
310316
sampling_params = SamplingParams(**params)
311317

0 commit comments

Comments
 (0)