From 96efb91480cd973dbcffab25ccb4b3a119b9929e Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Mon, 16 Mar 2026 00:35:49 -0700 Subject: [PATCH] [Model Runner V2] Fix processed logits in sample() (#37144) Signed-off-by: Woosuk Kwon --- vllm/v1/worker/gpu/sample/sampler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/v1/worker/gpu/sample/sampler.py b/vllm/v1/worker/gpu/sample/sampler.py index ec0087d9c..6f73ca87a 100644 --- a/vllm/v1/worker/gpu/sample/sampler.py +++ b/vllm/v1/worker/gpu/sample/sampler.py @@ -178,4 +178,4 @@ class Sampler: pos, apply_temperature=False, ) - return sampled, logits + return sampled, processed_logits