**Description** Bug fix to sync latest megatron-lm code.
Attach a file by drag & drop or click to upload
