Skip to content

Commit

Permalink
Update hifigan_generator.py
Browse files Browse the repository at this point in the history
fixing error
```
ImportError: cannot import name 'weight_norm' from 'torch.nn.utils.parametrizations'
```
module is in torch.nn.utils.weight_norm
  • Loading branch information
Sato942 authored Feb 18, 2024
1 parent dbf1a08 commit f69056d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion TTS/vocoder/models/hifigan_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit f69056d

Please sign in to comment.