Signed-off-by: Woosuk Kwon <woosuk@thinkingmachines.ai>
This commit is contained in:
Woosuk Kwon 2025-09-19 23:55:46 +00:00
parent a66aa37f40
commit 98ef239486

View File

@ -18,14 +18,14 @@ class SamplingMetadata:
temperature: torch.Tensor temperature: torch.Tensor
top_p: Optional[torch.Tensor] top_p: torch.Tensor | None
top_k: Optional[torch.Tensor] top_k: torch.Tensor | None
seeds: torch.Tensor seeds: torch.Tensor
pos: torch.Tensor pos: torch.Tensor
# None means no logprobs, 0 means sampled token logprobs only # None means no logprobs, 0 means sampled token logprobs only
max_num_logprobs: Optional[int] max_num_logprobs: int | None
@classmethod @classmethod
def make_dummy( def make_dummy(