游雁
2024-06-12 f6cae2b48b04ee1b64fcc1117087e60018d60a1e
funasr/models/llm_asr/adaptor.py
@@ -125,6 +125,7 @@
        olens = None
        olens = (ilens - 1) // self.k + 1
        masks = (~make_pad_mask(olens)[:, None, :]).to(x.device)
        if self.blocks is not None:
            for layer, block in enumerate(self.blocks):
                x, masks = block(x, masks)