From c029c4721ed93a9c364ae8bd4763e8fe5bce7340 Mon Sep 17 00:00:00 2001 From: Huy Vu2 Date: Thu, 9 May 2024 09:41:24 -0700 Subject: [PATCH] remove @deprecate import in megatron_retrieval_model.py --- .../nlp/models/language_modeling/megatron_retrieval_model.py | 1 - 1 file changed, 1 deletion(-) 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