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

Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>

[Model Runner V2] Fix processed logits in sample() (#37144)
Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>
96efb914 · Woosuk Kwon · GitHub · 2754231b · 96efb914
Unverified Commit 96efb914 authored Mar 16, 2026 by Woosuk Kwon Committed by GitHub Mar 16, 2026
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

vllm/v1/worker/gpu/sample/sampler.py vllm/v1/worker/gpu/sample/sampler.py +1 -1

No files found.
--- 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