Skip to content

Commit 1c69636

Browse files
jaywonchungAlvant
authored andcommitted
[Bugfix] Fix vocab_size field access in llava_next.py (vllm-project#6624)
Signed-off-by: Alvant <alvasian@yandex.ru>
1 parent f414f2e commit 1c69636

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

vllm/model_executor/models/llava.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,8 @@ def __init__(self,
155155
quant_config=quant_config)
156156
logit_scale = getattr(config, "logit_scale", 1.0)
157157
self.logits_processor = LogitsProcessor(self.unpadded_vocab_size,
158-
config.vocab_size, logit_scale)
158+
config.text_config.vocab_size,
159+
logit_scale)
159160
self.sampler = Sampler()
160161

161162
def _validate_pixel_values(self, data: torch.Tensor) -> torch.Tensor:

vllm/model_executor/models/llava_next.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,8 @@ def __init__(self,
249249
quant_config=quant_config)
250250
logit_scale = getattr(config, "logit_scale", 1.0)
251251
self.logits_processor = LogitsProcessor(self.unpadded_vocab_size,
252-
config.vocab_size, logit_scale)
252+
config.text_config.vocab_size,
253+
logit_scale)
253254
self.sampler = Sampler()
254255

255256
self.image_newline = nn.Parameter(

0 commit comments

Comments
 (0)