[Dev] fix(layer_wise): tag MTP-stage word_embeddings as is_embedding_or_output_parameter - #5180

[Dev] fix(layer_wise): tag MTP-stage word_embeddings as is_embedding_or_output_parameter#5180

Wohox wants to merge 1 commit into

NVIDIA:devNVIDIA/Megatron-LM:devfrom

Wohox:wohox/tag-mtp-word-embeddings-is-embedding-paramWohox/Megatron-LM:wohox/tag-mtp-word-embeddings-is-embedding-paramCopy head branch name to clipboard