雾聪
2024-01-15 8a8d60d5786510ec7b1dd4f622e848de8a15f8a8
funasr/frontends/wav_frontend.py
@@ -402,8 +402,7 @@
        self, input: torch.Tensor, input_lengths: torch.Tensor, cache: dict = {}, **kwargs
    ):
        is_final = kwargs.get("is_final", False)
        reset = kwargs.get("reset", False)
        if len(cache) == 0 or reset:
        if len(cache) == 0:
            self.init_cache(cache)
        
        batch_size = input.shape[0]