| | |
|---|
| | | from funasr.models.sanm.positionwise_feed_forward import PositionwiseFeedForwardDecoderSANM |
|---|
| | | from funasr.models.transformer.utils.repeat import repeat |
|---|
| | | |
|---|
| | | from funasr.utils.register import register_class, registry_tables |
|---|
| | | from funasr.register import tables |
|---|
| | | |
|---|
| | | class DecoderLayerSANM(nn.Module): |
|---|
| | | """Single decoder layer module. |
|---|
| | |
|---|
| | | return x, memory, fsmn_cache, opt_cache |
|---|
| | | |
|---|
| | | |
|---|
| | | @register_class("decoder_classes", "ParaformerSANMDecoder") |
|---|
| | | @tables.register("decoder_classes", "ParaformerSANMDecoder") |
|---|
| | | class ParaformerSANMDecoder(BaseTransformerDecoder): |
|---|
| | | """ |
|---|
| | | Author: Speech Lab of DAMO Academy, Alibaba Group |
|---|
