Merge pull request #388 from mittimithai/patch-3

Small fix for "Tacotron" use
2026-09-09 11:16:00 +08:00 · 2020-04-02 13:07:17 +02:00
parent 141797b6ae 6501369b0a
commit 99be88c338
1 changed files with 1 additions and 1 deletions
@@ -47,7 +47,7 @@ def setup_loader(ap, r, is_val=False, verbose=False):
        dataset = MyDataset(
            r,
            c.text_cleaner,
-            compute_linear_spec=True if c.model.lower() is 'tacotron' else False,
+            compute_linear_spec=True if c.model in ["Tacotron"] else False,
            meta_data=meta_data_eval if is_val else meta_data_train,
            ap=ap,
            tp=c.characters if 'characters' in c.keys() else None,