diff --git a/vllm/model_executor/models/granitemoehybrid.py b/vllm/model_executor/models/granitemoehybrid.py index 500ef1a1d041..19013cb87349 100644 --- a/vllm/model_executor/models/granitemoehybrid.py +++ b/vllm/model_executor/models/granitemoehybrid.py @@ -378,7 +378,7 @@ def forward( hidden_states = inputs_embeds else: hidden_states = self.embed_input_ids(input_ids) - hidden_states = hidden_states * self.embedding_multiplier + hidden_states = hidden_states * self.embedding_multiplier residual = None else: if intermediate_tensors is None: