Skip to content

Commit

Permalink
[Bugfix] Fix reinit procedure in ModelInputForGPUBuilder (vllm-projec…
Browse files Browse the repository at this point in the history
  • Loading branch information
alexm-neuralmagic authored Aug 9, 2024
1 parent fc7b8d1 commit 74af2bb
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions vllm/worker/model_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -348,6 +348,8 @@ def __init__(
self.multi_modal_inputs = multi_modal_inputs
self.prefix_cache_hit = prefix_cache_hit

self.n_seqs = len(self.seq_ids)

if not reinit:
self.__post_init__()

Expand Down

0 comments on commit 74af2bb

Please sign in to comment.