Dev gzf (#1440)

zhifu gao

2024-03-07 341182c3bfc62831aa02781d0e6bbe2a479f3fb2

 funasr/frontends/wav_frontend.py

@@ -399,11 +399,11 @@
        return feats_pad, feats_lens, lfr_splice_frame_idxs

    def forward(
        self, input: torch.Tensor, input_lengths: torch.Tensor, cache: dict = {}, **kwargs
        self, input: torch.Tensor, input_lengths: torch.Tensor, **kwargs
    ):
        is_final = kwargs.get("is_final", False)
        reset = kwargs.get("reset", False)
        if len(cache) == 0 or reset:
        cache = kwargs.get("cache", {})
        if len(cache) == 0:
            self.init_cache(cache)
        
        batch_size = input.shape[0]