xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-31 03:17:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

minor

Browse Source 
Signed-off-by: Woosuk Kwon <woosuk@thinkingmachines.ai>
This commit is contained in:
Woosuk Kwon 2025-09-19 19:07:46 +00:00
parent 396bbe67d3
commit 010e39ec7d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/v1/worker/gpu/sampler.py
View File

@ -17,10 +17,10 @@ class Sampler(nn.Module):
def __init__( def __init__(
self, self,
logprobs_mode: LogprobsMode = LogprobsMode.PROCESSED_LOGPROBS, logprobs_mode: LogprobsMode = "processed_logprobs",
): ):
super().__init__() super().__init__()
assert logprobs_mode == LogprobsMode.PROCESSED_LOGPROBS assert logprobs_mode == "processed_logprobs"
self.logprobs_mode = logprobs_mode self.logprobs_mode = logprobs_mode
def forward( def forward(