nichongjia-2007
2023-07-14 d14bb0f843aea0aeb254a5b3c21d42d04e28765b
funasr/modules/beam_search/beam_search_transducer.py
@@ -6,7 +6,6 @@
import numpy as np
import torch
from funasr.models.rnnt_predictor.abs_decoder import AbsDecoder
from funasr.models.joint_net.joint_network import JointNetwork
@@ -68,7 +67,7 @@
    def __init__(
        self,
        decoder: AbsDecoder,
        decoder,
        joint_network: JointNetwork,
        beam_size: int,
        lm: Optional[torch.nn.Module] = None,