Fix wrong argument name 🛠️

coqui-ai · Jun 28, 2021 · 9790edd · 9790edd
1 parent 7c61452
commit 9790edd
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/TTS/bin/compute_attention_masks.py b/TTS/bin/compute_attention_masks.py
@@ -89,7 +89,7 @@
         compute_linear_spec=False,
         ap=ap,
         meta_data=meta_data,
-        tp=C.characters if "characters" in C.keys() else None,
+        characters=c.characters if "characters" in C.keys() else None,
         add_blank=C["add_blank"] if "add_blank" in C.keys() else False,
         use_phonemes=C.use_phonemes,
         phoneme_cache_path=C.phoneme_cache_path,

diff --git a/TTS/bin/extract_tts_spectrograms.py b/TTS/bin/extract_tts_spectrograms.py
@@ -27,7 +27,7 @@ def setup_loader(ap, r, verbose=False):
         compute_linear_spec=False,
         meta_data=meta_data,
         ap=ap,
-        tp=c.characters if "characters" in c.keys() else None,
+        characters=c.characters if "characters" in c.keys() else None,
         add_blank=c["add_blank"] if "add_blank" in c.keys() else False,
         batch_group_size=0,
         min_seq_len=c.min_seq_len,

diff --git a/TTS/tts/models/base_tts.py b/TTS/tts/models/base_tts.py
@@ -164,7 +164,7 @@ def get_data_loader(
                 compute_linear_spec=config.model.lower() == "tacotron",
                 meta_data=data_items,
                 ap=ap,
-                tp=config.characters,
+                characters=config.characters,
                 add_blank=config["add_blank"],
                 batch_group_size=0 if is_eval else config.batch_group_size * config.batch_size,
                 min_seq_len=config.min_seq_len,

diff --git a/notebooks/ExtractTTSpectrogram.ipynb b/notebooks/ExtractTTSpectrogram.ipynb
@@ -112,7 +112,7 @@
     "preprocessor = importlib.import_module('TTS.tts.datasets.preprocess')\n",
     "preprocessor = getattr(preprocessor, DATASET.lower())\n",
     "meta_data = preprocessor(DATA_PATH,METADATA_FILE)\n",
-    "dataset = TTSDataset(checkpoint['r'], C.text_cleaner, False, ap, meta_data,tp=C.characters if 'characters' in C.keys() else None, use_phonemes=C.use_phonemes,  phoneme_cache_path=C.phoneme_cache_path, enable_eos_bos=C.enable_eos_bos_chars)\n",
+    "dataset = TTSDataset(checkpoint['r'], C.text_cleaner, False, ap, meta_data,characters=c.characters if 'characters' in C.keys() else None, use_phonemes=C.use_phonemes,  phoneme_cache_path=C.phoneme_cache_path, enable_eos_bos=C.enable_eos_bos_chars)\n",
     "loader = DataLoader(dataset, batch_size=BATCH_SIZE, num_workers=4, collate_fn=dataset.collate_fn, shuffle=False, drop_last=False)"
    ]
   },

diff --git a/tests/data_tests/test_loader.py b/tests/data_tests/test_loader.py
@@ -44,7 +44,7 @@ def _create_dataloader(self, batch_size, r, bgs):
             compute_linear_spec=True,
             ap=self.ap,
             meta_data=items,
-            tp=c.characters,
+            characters=c.characters,
             batch_group_size=bgs,
             min_seq_len=c.min_seq_len,
             max_seq_len=float("inf"),