Merge pull request #1144 from yeyupiaoling/main

Shi Xian

2023-12-06 b3fcd42bf6e60372ed0f0d3e53798fcdd713fe08

 funasr/modules/beam_search/beam_search_transducer.py

@@ -6,8 +6,7 @@
import numpy as np
import torch

from funasr.models.rnnt_decoder.abs_decoder import AbsDecoder
from funasr.models.joint_network import JointNetwork
from funasr.models.joint_net.joint_network import JointNetwork


@dataclass
@@ -68,7 +67,7 @@

    def __init__(
        self,
        decoder: AbsDecoder,
        decoder,
        joint_network: JointNetwork,
        beam_size: int,
        lm: Optional[torch.nn.Module] = None,