游雁
2024-06-13 f97f3e8dd5b40be6faaccd9c089aea52ceba29ef
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)