Fix the weight_norm import issue (#195)

Co-authored-by: 80268185 <[email protected]>
wenet-e2e · Jan 17, 2024 · 7e4820f · 7e4820f · Jackiexiao · Jan 23, 2024
1 parent c5ec709
commit 7e4820f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/wetts/vits/model/decoders.py b/wetts/vits/model/decoders.py
@@ -3,7 +3,7 @@
 from torch.nn import Conv1d, ConvTranspose1d
 from torch.nn import functional as F
 from torch.nn.utils import remove_weight_norm
-from torch.nn.utils.parametrizations import weight_norm
+from torch.nn.utils import weight_norm
 from torchaudio.transforms import InverseSpectrogram
 
 from model.modules import LRELU_SLOPE

diff --git a/wetts/vits/model/discriminators.py b/wetts/vits/model/discriminators.py
@@ -5,7 +5,7 @@
 from torch.nn import functional as F
 from torch.nn import Conv1d, Conv2d
 from torch.nn.utils import spectral_norm
-from torch.nn.utils.parametrizations import weight_norm
+from torch.nn.utils import weight_norm
 from torchaudio.transforms import Spectrogram
 
 from model.modules import LRELU_SLOPE

diff --git a/wetts/vits/model/modules.py b/wetts/vits/model/modules.py
@@ -1,6 +1,6 @@
 import torch
 from torch import nn
-from torch.nn.utils.parametrizations import weight_norm
+from torch.nn.utils import weight_norm
 from torch.nn.utils.parametrize import remove_parametrizations
 
 from utils import commons