add get_output_embeddings()

#65

by ranchlai - opened Jul 26, 2023

Jul 26, 2023

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.

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

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

· Sign up or log in to comment