mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-13 09:19:11 +08:00
fix
Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
parent
efba25e21a
commit
e451045a66
@ -2270,6 +2270,8 @@ class GPUModelRunner(LoRAModelRunnerMixin, KVConnectorModelRunnerMixin):
|
|||||||
bad_words_token_ids={},
|
bad_words_token_ids={},
|
||||||
logitsprocs=LogitsProcessors(),
|
logitsprocs=LogitsProcessors(),
|
||||||
token_ids=None,
|
token_ids=None,
|
||||||
|
num_tokens=None,
|
||||||
|
num_prompt_tokens=None,
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
sampler_output = self.sampler(logits=logits,
|
sampler_output = self.sampler(logits=logits,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user