small fix for class name matching

pull/10/head
erogol 2020-07-08 10:25:56 +02:00
parent 7dcdaa14ab
commit 512188469b
1 changed files with 1 additions and 1 deletions

View File

@ -91,7 +91,7 @@ def trim_silence(wav, ap):
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)
else:
wav = ap.inv_melspectrogram(postnet_output.T)