From 512188469b1c943ffd47c162f496c44b997b441b Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Wed, 8 Jul 2020 10:25:56 +0200
Subject: [PATCH] small fix for class name matching

---
 utils/synthesis.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/synthesis.py b/utils/synthesis.py
index a53c12dc..03d7072e 100644
--- a/utils/synthesis.py
+++ b/utils/synthesis.py
@@ -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)