Update modeling_mobilellm.py

modeling_mobilellm.py

@@ -1145,9 +1145,9 @@ class MobileLLMForCausalLM(LlamaPreTrainedModel):
 
     def get_output_embeddings(self):
         return (
-            self.lm_head
-            if not getattr(self.config, "share_embedding", False)
-            else self.get_input_embeddings()
+            self.get_input_embeddings()
+            if getattr(self.config, "share_embedding", True)
+            else self.lm_head
         )
 
     def set_output_embeddings(self, new_embeddings):