Fix importerror in transformer_lm_model after nlp module removals (NVIDIA-NeMo#14199)

chtruong814 · nasretdinovr · commit c273ae3ed626 · 2025-08-08T08:50:08.000-07:00
Signed-off-by: Charlie Truong &lt;chtruong@nvidia.com&gt;
diff --git a/nemo/collections/nlp/models/language_modeling/transformer_lm_model.py b/nemo/collections/nlp/models/language_modeling/transformer_lm_model.py
@@ -31,7 +31,7 @@
 
 try:
     from nemo.collections.nlp.metrics import SequencePerplexity
-except ModuleNotFoundError:
+except (ImportError, ModuleNotFoundError):
     from abc import ABC
 
     SequencePerplexity = ABC
