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.
· Sign up or log in to comment
