aky15
2023-04-14 fa25b637b0d257186a8399eb1c530a91f4252702
funasr/tasks/asr_transducer.py
@@ -21,10 +21,10 @@
    LightweightConvolutionTransformerDecoder,
    TransformerDecoder,
)
from funasr.models.rnnt_decoder.abs_decoder import AbsDecoder
from funasr.models.rnnt_decoder.rnn_decoder import RNNDecoder
from funasr.models.rnnt_decoder.stateless_decoder import StatelessDecoder
from funasr.models.encoder.chunk_encoder import ChunkEncoder as Encoder
from funasr.models.rnnt_predictor.abs_decoder import AbsDecoder
from funasr.models.rnnt_predictor.rnn_decoder import RNNDecoder
from funasr.models.rnnt_predictor.stateless_decoder import StatelessDecoder
from funasr.models.encoder.conformer_encoder import ConformerChunkEncoder
from funasr.models.e2e_transducer import TransducerModel
from funasr.models.e2e_transducer_unified import UnifiedTransducerModel
from funasr.models.joint_network import JointNetwork
@@ -72,9 +72,9 @@
encoder_choices = ClassChoices(
        "encoder",
        classes=dict(
                encoder=Encoder,
                chunk_conformer=ConformerChunkEncoder,
        ),
        default="encoder",
        default="chunk_conformer",
)
decoder_choices = ClassChoices(