游雁
2024-06-11 f57b3788f2c91e52a48629423c7bc0539e44f793
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)