remove @deprecate import in megatron_retrieval_model.py

NVIDIA · May 9, 2024 · c029c47 · c029c47
1 parent 7d6d511
commit c029c47
Showing 1 changed file with 0 additions and 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
@@ -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