diff --git a/vllm/model_executor/models/adapters.py b/vllm/model_executor/models/adapters.py index cdbf628e610..03cd55cf100 100644 --- a/vllm/model_executor/models/adapters.py +++ b/vllm/model_executor/models/adapters.py @@ -539,7 +539,7 @@ def load_weights_using_from_2_way_softmax( text_config.vocab_size, text_config.hidden_size, ) - if text_config.tie_word_embeddings: + if language_model.config.get_text_config().tie_word_embeddings: # embed_tokens is the assumed name for input embeddings. If the model does not # have this attribute, we fall back to get_input_embeddings(), which is used by # the Transformers modeling backend. @@ -608,7 +608,7 @@ def load_weights_no_post_processing(model, weights: Iterable[tuple[str, torch.Te text_config.vocab_size, text_config.hidden_size, ) - if text_config.tie_word_embeddings: + if language_model.config.get_text_config().tie_word_embeddings: # embed_tokens is the assumed name for input embeddings. If the model does not # have this attribute, we fall back to get_input_embeddings(), which is used by # the Transformers modeling backend.