xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-06-01 16:07:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

ruff change

Browse Source 
Signed-off-by: qli88 <qiang.li2@amd.com>
This commit is contained in:
qli88 2025-11-21 15:58:03 +00:00
parent eea8b6634c
commit a1033a9542
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/samplers/test_logprobs.py
View File

@ -25,7 +25,9 @@ def test_ranks(
flat_logprobs, flat_logprobs,
example_prompts, example_prompts,
): ):
with vllm_runner(model, dtype=dtype, max_logprobs=MAX_LOGPROBS, enforce_eager=True) as vllm_model: with vllm_runner(
model, dtype=dtype, max_logprobs=MAX_LOGPROBS, enforce_eager=True
) as vllm_model:
tokenizer = vllm_model.llm.get_tokenizer() tokenizer = vllm_model.llm.get_tokenizer()
example_prompt_tokens = [tokenizer.encode(prompt) for prompt in example_prompts] example_prompt_tokens = [tokenizer.encode(prompt) for prompt in example_prompts]
sampling_params = SamplingParams( sampling_params = SamplingParams(