| | |
|---|
| | | from funasr.models.transformer.utils.repeat import repeat |
|---|
| | | from funasr.models.paraformer.decoder import DecoderLayerSANM, ParaformerSANMDecoder |
|---|
| | | |
|---|
| | | from funasr.utils.register import register_class, registry_tables |
|---|
| | | from funasr.register import tables |
|---|
| | | |
|---|
| | | class ContextualDecoderLayer(nn.Module): |
|---|
| | | def __init__( |
|---|
| | |
|---|
| | | x = self.dropout(self.src_attn(x, memory, memory_mask)) |
|---|
| | | return x, tgt_mask, memory, memory_mask, cache |
|---|
| | | |
|---|
| | | @register_class("decoder_classes", "ContextualParaformerDecoder") |
|---|
| | | @tables.register("decoder_classes", "ContextualParaformerDecoder") |
|---|
| | | class ContextualParaformerDecoder(ParaformerSANMDecoder): |
|---|
| | | """ |
|---|
| | | Author: Speech Lab of DAMO Academy, Alibaba Group |
|---|
