soiz commited on
Commit
426f4f5
·
verified ·
1 Parent(s): c61cd8a

Update attentions.py

Browse files
Files changed (1) hide show
  1. attentions.py +1 -1
attentions.py CHANGED
@@ -2,7 +2,7 @@ import math
2
  import torch
3
  from torch import nn
4
  from torch.nn import functional as F
5
-
6
  import commons
7
  from modules import LayerNorm
8
 
 
2
  import torch
3
  from torch import nn
4
  from torch.nn import functional as F
5
+ from torch.nn.utils.parametrizations import weight_norm
6
  import commons
7
  from modules import LayerNorm
8