mirror of https://github.com/coqui-ai/TTS.git
Fix `num_char` init in Tacotron models
parent
e685ddfca7
commit
6af03ac476
|
@ -102,7 +102,8 @@
|
|||
"tacotron2-DDC-GST":{
|
||||
"github_rls_url": "https://github.com/coqui-ai/TTS/releases/download/v0.0.10/tts_models--zh-CN--baker--tacotron2-DDC-GST.zip",
|
||||
"commit": "unknown",
|
||||
"author": "@kirianguiller"
|
||||
"author": "@kirianguiller",
|
||||
"default_vocoder": null
|
||||
}
|
||||
}
|
||||
},
|
||||
|
|
|
@ -208,7 +208,7 @@ def main():
|
|||
if args.vocoder_name is not None and not args.vocoder_path:
|
||||
vocoder_path, vocoder_config_path, _ = manager.download_model(args.vocoder_name)
|
||||
|
||||
# CASE3: set custome model paths
|
||||
# CASE3: set custom model paths
|
||||
if args.model_path is not None:
|
||||
model_path = args.model_path
|
||||
config_path = args.config_path
|
||||
|
|
|
@ -24,6 +24,7 @@ class Tacotron(BaseTacotron):
|
|||
super().__init__(config)
|
||||
|
||||
self.num_chars, self.config = self.get_characters(config)
|
||||
config.num_chars = self.num_chars
|
||||
|
||||
# pass all config fields to `self`
|
||||
# for fewer code change
|
||||
|
|
|
@ -23,7 +23,7 @@ class Tacotron2(BaseTacotron):
|
|||
super().__init__(config)
|
||||
|
||||
chars, self.config = self.get_characters(config)
|
||||
self.num_chars = len(chars)
|
||||
config.num_chars = len(chars)
|
||||
self.decoder_output_dim = config.out_channels
|
||||
|
||||
# pass all config fields to `self`
|
||||
|
|
Loading…
Reference in New Issue