Update attentions.py
This commit is contained in:
@@ -7,7 +7,7 @@ from torch.nn import functional as F
|
|||||||
|
|
||||||
import commons
|
import commons
|
||||||
import modules
|
import modules
|
||||||
|
from torch.nn.utils import weight_norm, remove_weight_norm
|
||||||
class LayerNorm(nn.Module):
|
class LayerNorm(nn.Module):
|
||||||
def __init__(self, channels, eps=1e-5):
|
def __init__(self, channels, eps=1e-5):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|||||||
Reference in New Issue
Block a user