Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
Woosuk Kwon 2025-08-28 12:55:13 -07:00
parent efba25e21a
commit e451045a66

View File

@ -2270,6 +2270,8 @@ class GPUModelRunner(LoRAModelRunnerMixin, KVConnectorModelRunnerMixin):
bad_words_token_ids={},
logitsprocs=LogitsProcessors(),
token_ids=None,
num_tokens=None,
num_prompt_tokens=None,
)
try:
sampler_output = self.sampler(logits=logits,