Add __all__ and __init__ files for the conformer model (#14)

Author: Atticus1806

i6_models/__init__.py

@@ -0,0 +1,4 @@
+from .convolution import *
+from .feedforward import *
+from .mhsa import *
+from .norm import *

i6_models/parts/conformer/__init__.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+__all__ = ["ConformerConvolutionV1", "ConformerConvolutionV1Config"]
+
 from dataclasses import dataclass
 
 import torch

i6_models/parts/conformer/convolution.py

@@ -1,4 +1,7 @@
 from __future__ import annotations
+
+__all__ = ["ConformerPositionwiseFeedForwardV1", "ConformerPositionwiseFeedForwardV1Config"]
+
 from dataclasses import dataclass
 from typing import Callable
 

i6_models/parts/conformer/feedforward.py

@@ -1,6 +1,8 @@
 from __future__ import annotations
+
+__all__ = ["ConformerMHSAV1", "ConformerMHSAV1Config"]
 from dataclasses import dataclass
-from typing import Optional, Callable
+from typing import Optional
 import torch
 
 from i6_models.config import ModelConfiguration

i6_models/parts/conformer/mhsa.py

@@ -1,3 +1,4 @@
+__all__ = ["LayerNormNC"]
 import torch
 import torch.nn as nn