From c8b9ca3d714e5f475030e87bd12a9d56873a8b57 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 10 Aug 2021 08:56:34 +0000
Subject: [PATCH] Fix Tacotron num_char init

---
 TTS/tts/models/tacotron.py     | 4 ++--
 TTS/vocoder/models/__init__.py | 6 ++++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/TTS/tts/models/tacotron.py b/TTS/tts/models/tacotron.py
index f7969b19..84a256d5 100644
--- a/TTS/tts/models/tacotron.py
+++ b/TTS/tts/models/tacotron.py
@@ -23,8 +23,8 @@ class Tacotron(BaseTacotron):
     def __init__(self, config: Coqpit):
         super().__init__(config)
 
-        self.num_chars, self.config = self.get_characters(config)
-        config.num_chars = self.num_chars
+        chars, self.config = self.get_characters(config)
+        config.num_chars = self.num_chars = len(chars)
 
         # pass all config fields to `self`
         # for fewer code change
diff --git a/TTS/vocoder/models/__init__.py b/TTS/vocoder/models/__init__.py
index 7c209af4..edc94d72 100644
--- a/TTS/vocoder/models/__init__.py
+++ b/TTS/vocoder/models/__init__.py
@@ -24,8 +24,10 @@ def setup_model(config: Coqpit):
         elif config.model.lower() == "wavegrad":
             MyModel = getattr(MyModel, "Wavegrad")
         else:
-            MyModel = getattr(MyModel, to_camel(config.model))
-            raise ValueError(f"Model {config.model} not exist!")
+            try:
+                MyModel = getattr(MyModel, to_camel(config.model))
+            except ModuleNotFoundError as e:
+                raise ValueError(f"Model {config.model} not exist!") from e
     model = MyModel(config)
     return model