fix: import of weight_norm

wittech · Dec 11, 2023 · 354af0a · 354af0a
1 parent 39f21ca
commit 354af0a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/models/hifigan/models.py b/models/hifigan/models.py
@@ -7,7 +7,10 @@
 import torch.nn.functional as F
 import torch.nn as nn
 from torch.nn import Conv1d, ConvTranspose1d, AvgPool1d, Conv2d
-from torch.nn.utils.parametrizations import weight_norm
+try:
+ from torch.nn.utils.parametrizations import weight_norm
+except:
+ from torch.nn.utils import weight_norm
 from torch.nn.utils import remove_weight_norm, spectral_norm
 
 LRELU_SLOPE = 0.1