| | |
|---|
| | | |
|---|
| | | from funasr.models.ctc.ctc import CTC |
|---|
| | | |
|---|
| | | from funasr.utils.register import register_class, registry_tables |
|---|
| | | from funasr.register import tables |
|---|
| | | |
|---|
| | | class EncoderLayerSANM(nn.Module): |
|---|
| | | def __init__( |
|---|
| | |
|---|
| | | return x, cache |
|---|
| | | |
|---|
| | | |
|---|
| | | @register_class("encoder_classes", "SANMEncoderChunkOpt") |
|---|
| | | @tables.register("encoder_classes", "SANMEncoderChunkOpt") |
|---|
| | | class SANMEncoderChunkOpt(nn.Module): |
|---|
| | | """ |
|---|
| | | Author: Speech Lab of DAMO Academy, Alibaba Group |
|---|
| | |
|---|
| | | xs_pad: torch.Tensor, |
|---|
| | | ilens: torch.Tensor, |
|---|
| | | cache: dict = None, |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | is_final = kwargs.get("is_final", False) |
|---|
| | | xs_pad *= self.output_size() ** 0.5 |
|---|
| | | if self.embed is None: |
|---|
| | | xs_pad = xs_pad |
|---|
