From 3593ecad5588b7756a3a6732ed323f7a50e9fcbd Mon Sep 17 00:00:00 2001 From: Can Balioglu Date: Wed, 27 Sep 2023 22:13:20 -0400 Subject: [PATCH] Export LayerNorm (#70) --- src/fairseq2/nn/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/fairseq2/nn/__init__.py b/src/fairseq2/nn/__init__.py index 2798325a1..eed044d1a 100644 --- a/src/fairseq2/nn/__init__.py +++ b/src/fairseq2/nn/__init__.py @@ -8,6 +8,9 @@ from fairseq2.nn.incremental_state import IncrementalState as IncrementalState from fairseq2.nn.incremental_state import IncrementalStateBag as IncrementalStateBag from fairseq2.nn.module_list import ModuleList as ModuleList +from fairseq2.nn.normalization import LayerNorm as LayerNorm +from fairseq2.nn.normalization import RMSNorm as RMSNorm +from fairseq2.nn.normalization import StandardLayerNorm as StandardLayerNorm from fairseq2.nn.position_encoder import ( LearnedPositionEncoder as LearnedPositionEncoder, )