File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed
vllm/model_executor/models Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -155,7 +155,8 @@ def __init__(self,
155
155
quant_config = quant_config )
156
156
logit_scale = getattr (config , "logit_scale" , 1.0 )
157
157
self .logits_processor = LogitsProcessor (self .unpadded_vocab_size ,
158
- config .vocab_size , logit_scale )
158
+ config .text_config .vocab_size ,
159
+ logit_scale )
159
160
self .sampler = Sampler ()
160
161
161
162
def _validate_pixel_values (self , data : torch .Tensor ) -> torch .Tensor :
Original file line number Diff line number Diff line change @@ -249,7 +249,8 @@ def __init__(self,
249
249
quant_config = quant_config )
250
250
logit_scale = getattr (config , "logit_scale" , 1.0 )
251
251
self .logits_processor = LogitsProcessor (self .unpadded_vocab_size ,
252
- config .vocab_size , logit_scale )
252
+ config .text_config .vocab_size ,
253
+ logit_scale )
253
254
self .sampler = Sampler ()
254
255
255
256
self .image_newline = nn .Parameter (
You can’t perform that action at this time.
0 commit comments