Update hifigan_generator.py

fixing error
```
ImportError: cannot import name 'weight_norm' from 'torch.nn.utils.parametrizations'
```
module is in torch.nn.utils.weight_norm
This commit is contained in:
Mubarak Harran Alketbi 2024-02-18 22:46:30 +04:00 committed by GitHub
parent dbf1a08a0d
commit f69056d653
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -3,7 +3,7 @@ import torch
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