Unverified Commit e631f8e7 authored by Gabe Goodhart's avatar Gabe Goodhart Committed by GitHub
Browse files

fix: Apply embedding_multiplier to inputs_embeds (#34813)


Signed-off-by: default avatarGabe Goodhart <ghart@us.ibm.com>
Co-authored-by: default avatarCyrus Leung <cyrus.tl.leung@gmail.com>
Co-authored-by: default avatarCyrus Leung <tlleungac@connect.ust.hk>
parent e97c46a9
......@@ -378,7 +378,7 @@ class GraniteMoeHybridModel(nn.Module):
hidden_states = inputs_embeds
else:
hidden_states = self.embed_input_ids(input_ids)
hidden_states = hidden_states * self.embedding_multiplier
hidden_states *= self.embedding_multiplier
residual = None
else:
if intermediate_tensors is None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment