diff --git a/models/attn.py b/models/attn.py
index d8588b1..6ca983c 100644
--- a/models/attn.py
+++ b/models/attn.py
@@ -1,9 +1,4 @@
 
-import numpy as np
-
-from math import sqrt
-from utils.masking import TriangularCausalMask, ProbMask
-
 import numpy as np
 import mindspore.nn as nn
 import mindspore.ops as ops