minor

Signed-off-by: Woosuk Kwon <woosuk@thinkingmachines.ai>
2025-09-19 19:07:46 +00:00
parent 396bbe67d3
commit 010e39ec7d
1 changed files with 2 additions and 2 deletions
--- 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(