MCPcopy Create free account
hub / github.com/OpenMOSS/MOSS / set_output_embeddings

Method set_output_embeddings

models/modeling_moss.py:619–620  ·  view source on GitHub ↗
(self, new_embeddings)

Source from the content-addressed store, hash-verified

617 return self.lm_head
618
619 def set_output_embeddings(self, new_embeddings):
620 self.lm_head = new_embeddings
621
622 def prepare_inputs_for_generation(self, input_ids, past_key_values=None, **kwargs):
623 token_type_ids = kwargs.get("token_type_ids", None)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected