**Describe the bug** detailed information please refer to https://github.com/NVIDIA/NeMo/pull/7627