Spaces:
Running on Zero
Running on Zero
| from .attention import CompactRelPositionalEncoding, RelPositionMultiheadAttentionWeights, SelfAttention | |
| from .downsampling import DownsampledZipformer2Encoder, SimpleDownsample, SimpleUpsample | |
| from .encoder import Zipformer2 | |
| from .encoder_layer import BypassModule, Zipformer2EncoderLayer | |
| from .encoder_stack import Zipformer2Encoder | |
| from .modules import ConvolutionModule, FeedforwardModule, NonlinAttention | |
| __all__ = [ | |
| "BypassModule", | |
| "CompactRelPositionalEncoding", | |
| "ConvolutionModule", | |
| "DownsampledZipformer2Encoder", | |
| "FeedforwardModule", | |
| "NonlinAttention", | |
| "RelPositionMultiheadAttentionWeights", | |
| "SelfAttention", | |
| "SimpleDownsample", | |
| "SimpleUpsample", | |
| "Zipformer2", | |
| "Zipformer2Encoder", | |
| "Zipformer2EncoderLayer", | |
| ] | |