Merge branch 'torch_norm_alias' into 'main'

Add TorchLayerNorm alias for backward compatibility See merge request ADLR/megatron-lm!2391
NVIDIA · Nov 29, 2024 · 31a29b8 · 31a29b8
2 parents 8e9d4dc + e842d46
commit 31a29b8
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