[Model Runner V2] Fix processed logits in sample() (#37144)

Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>
2026-03-16 00:35:49 -07:00
parent 2754231ba3
commit 96efb91480
1 changed files with 1 additions and 1 deletions
--- 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