ADLR/megatron-lm!2391 - Add TorchLayerNorm alias for backward compati…

…bility
NVIDIA · Nov 29, 2024 · e842d46 · e842d46
1 parent 081ab4d
commit e842d46
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/megatron/core/transformer/torch_layer_norm.py b/megatron/core/transformer/torch_layer_norm.py
@@ -0,0 +1,4 @@
+# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
+from megatron.core.transformer.torch_norm import WrappedTorchNorm
+
+WrappedTorchLayerNorm = WrappedTorchNorm