(self, B, L, device="cpu")
| 25 | |
| 26 | class TriangularCausalMask(): |
| 27 | def __init__(self, B, L, device="cpu"): |
| 28 | mask_shape = [B, 1, L, L] |
| 29 | with torch.no_grad(): |
| 30 | self._mask = torch.triu(torch.ones(mask_shape, dtype=torch.bool), diagonal=1).to(device) |
| 31 | |
| 32 | @property |
| 33 | def mask(self): |
nothing calls this directly
no outgoing calls
no test coverage detected