diff --git a/nemo/collections/asr/parts/conformer_modules.py b/nemo/collections/asr/parts/conformer_modules.py index 212b6ce4248c..ef8943d4d510 100644 --- a/nemo/collections/asr/parts/conformer_modules.py +++ b/nemo/collections/asr/parts/conformer_modules.py @@ -19,7 +19,7 @@ from nemo.collections.asr.parts.activations import Swish from nemo.collections.asr.parts.multi_head_attention import MultiHeadAttention, RelPositionMultiHeadAttention -__all__ = ['ConformerConvolution', 'ConformerFeedForward'] +__all__ = ['ConformerConvolution', 'ConformerFeedForward', 'ConformerEncoderBlock'] class ConformerEncoderBlock(torch.nn.Module):