Yunshui Li

Results 1 issues of Yunshui Li

```python cls_mask = rearrange(text!=self.pad_id, 'b j -> b 1 j') attn_mask = F.pad(cls_mask, (0, 1, seq, 0), value=True) attn_mask = rearrange(attn_mask, 'b i j -> b 1 i j') sim...