decoding

游雁

2024-05-06 b139b9cfaef2f69b92dc9f59d55216132d690e38

decoding

 funasr/models/sense_voice/decoder.py

@@ -472,7 +472,7 @@
        is_pad_mask = kwargs.get("is_pad_mask", False)
        is_pad_memory_mask = kwargs.get("is_pad_memory_mask", False)

        fsmn_cache = cache[layer]["fsmn_cache"] if len(cache) > 0 else None
        fsmn_cache = cache[layer]["fsmn_cache"] if len(cache) > 0 or cache is None else None
        # if fsmn_cache is not None:
        #     x = x[:, -1:]
        att_res, fsmn_cache = self.attn(self.attn_ln(x), mask=None, cache=fsmn_cache)