From 010e39ec7d6c75a4e3f2e140148d271598ec9c37 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Fri, 19 Sep 2025 19:07:46 +0000 Subject: [PATCH] minor Signed-off-by: Woosuk Kwon --- vllm/v1/worker/gpu/sampler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/v1/worker/gpu/sampler.py b/vllm/v1/worker/gpu/sampler.py index d03309abca61a..4b4a250ebaa19 100644 --- a/vllm/v1/worker/gpu/sampler.py +++ b/vllm/v1/worker/gpu/sampler.py @@ -17,10 +17,10 @@ class Sampler(nn.Module): def __init__( self, - logprobs_mode: LogprobsMode = LogprobsMode.PROCESSED_LOGPROBS, + logprobs_mode: LogprobsMode = "processed_logprobs", ): super().__init__() - assert logprobs_mode == LogprobsMode.PROCESSED_LOGPROBS + assert logprobs_mode == "processed_logprobs" self.logprobs_mode = logprobs_mode def forward(