megatron_gpt_test.py throws ImportError

[yzhang123]

NeMo/examples/nlp/language_modeling$ python megatron_gpt_test.py

throws

from nemo.collections.nlp.parts.nlp_overrides import (
ImportError: cannot import name 'NLPMixedPrecisionPlugin' from 'nemo.collections.nlp.parts.nlp_overrides

Environment details

If NVIDIA docker image is used you don't need to specify these.
Otherwise, please provide:

• OS version
• main

Additional context

Add any other context about the problem here.
Example: GPU model