diff --git a/vllm/v1/worker/gpu/sample/sampler.py b/vllm/v1/worker/gpu/sample/sampler.py index ec0087d9c8b1..6f73ca87ac67 100644 --- a/vllm/v1/worker/gpu/sample/sampler.py +++ b/vllm/v1/worker/gpu/sample/sampler.py @@ -178,4 +178,4 @@ def sample( pos, apply_temperature=False, ) - return sampled, logits + return sampled, processed_logits