diff --git a/nemo/collections/nlp/models/language_modeling/megatron_retrieval_model.py b/nemo/collections/nlp/models/language_modeling/megatron_retrieval_model.py index d7167eeae3ce..42323e503f7d 100644 --- a/nemo/collections/nlp/models/language_modeling/megatron_retrieval_model.py +++ b/nemo/collections/nlp/models/language_modeling/megatron_retrieval_model.py @@ -56,7 +56,6 @@ ) from nemo.collections.nlp.parts.nlp_overrides import GradScaler from nemo.utils import AppState, logging -from nemo.utils.decorators import deprecated try: from megatron.core import parallel_state