diff --git a/utils/masking.py b/utils/masking.py index 6e37408..6d58992 100644 --- a/utils/masking.py +++ b/utils/masking.py @@ -2,7 +2,6 @@ import mindspore import mindspore.ops.operations as P import mindspore.numpy as mnp - class TriangularCausalMask(): def __init__(self, B, L, device="cpu"): mask_shape = [B, 1, L, L]