mirror of https://github.com/coqui-ai/TTS.git
change import for normalization layer
This commit is contained in:
parent
c008003506
commit
e4c6386603
|
@ -3,7 +3,8 @@ from torch import nn
|
|||
from torch.nn import functional as F
|
||||
|
||||
from TTS.tts.utils.generic_utils import sequence_mask
|
||||
from TTS.tts.layers.glow_tts.glow import InvConvNear, CouplingBlock, ActNorm
|
||||
from TTS.tts.layers.glow_tts.glow import InvConvNear, CouplingBlock
|
||||
from TTS.tts.layers.glow_tts.normalization import ActNorm
|
||||
|
||||
|
||||
def squeeze(x, x_mask=None, num_sqz=2):
|
||||
|
|
Loading…
Reference in New Issue