add get_output_embeddings()

#65
by ranchlai - opened

Add(override) the get_output_embeddings() in model class to return the correct LM_head module.
This is useful for applications that need to find out the name of LM_head module.

Your need to confirm your account before you can post a new comment.

Sign up or log in to comment