Address import error exception issue

Signed-off-by: John St John <[email protected]>
NVIDIA · Feb 21, 2025 · 04eba8e · 04eba8e
1 parent 4480311
commit 04eba8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/llm/gpt/model/megatron/hyena/hyena_mixer.py b/nemo/collections/llm/gpt/model/megatron/hyena/hyena_mixer.py
@@ -44,7 +44,7 @@
 
 try:
     from transformer_engine.common.recipe import DelayedScaling, Format
-except:
+except ImportError:
     logger.warning("WARNING: transformer_engine not installed. Using default recipe.")