mirror of https://github.com/coqui-ai/TTS.git
small fix for class name matching
This commit is contained in:
parent
7dcdaa14ab
commit
512188469b
|
@ -91,7 +91,7 @@ def trim_silence(wav, ap):
|
||||||
|
|
||||||
|
|
||||||
def inv_spectrogram(postnet_output, ap, CONFIG):
|
def inv_spectrogram(postnet_output, ap, CONFIG):
|
||||||
if CONFIG.model in ["Tacotron", "TacotronGST"]:
|
if CONFIG.model.lower() in ["tacotron"]:
|
||||||
wav = ap.inv_spectrogram(postnet_output.T)
|
wav = ap.inv_spectrogram(postnet_output.T)
|
||||||
else:
|
else:
|
||||||
wav = ap.inv_melspectrogram(postnet_output.T)
|
wav = ap.inv_melspectrogram(postnet_output.T)
|
||||||
|
|
Loading…
Reference in New Issue