diff --git a/attentions.py b/attentions.py
index 4a42a1a..ec82ded 100644
--- a/attentions.py
+++ b/attentions.py
@@ -7,7 +7,7 @@ from torch.nn import functional as F
 
 import commons
 import modules
-
+from torch.nn.utils import weight_norm, remove_weight_norm
 class LayerNorm(nn.Module):
   def __init__(self, channels, eps=1e-5):
     super().__init__()