Fix regex capture for Megatron KD PP layer renaming#1355

AAnoosheh merged 2 commits into

mainNVIDIA/Model-Optimizer:mainfrom

aanoosheh/safer-megatron-pp-regexNVIDIA/Model-Optimizer:aanoosheh/safer-megatron-pp-regexCopy head branch name to clipboard