diff --git a/TTS/vocoder/models/hifigan_generator.py b/TTS/vocoder/models/hifigan_generator.py index 9247532259..e1faa4e7c6 100644 --- a/TTS/vocoder/models/hifigan_generator.py +++ b/TTS/vocoder/models/hifigan_generator.py @@ -3,7 +3,7 @@ from torch import nn from torch.nn import Conv1d, ConvTranspose1d from torch.nn import functional as F -from torch.nn.utils.parametrizations import weight_norm +from torch.nn.utils.weight_norm import weight_norm from torch.nn.utils.parametrize import remove_parametrizations from TTS.utils.io import load_fsspec