zhifu gao
2024-06-11 20aa07268a7fafaaab7762b488615af32a0e82b4
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)