Merge pull request #6 from kirianguiller/dev-pr3

Dev pr3 : small changes for chinese implementation
coqui-ai · Mar 2, 2021 · d706d4c · d706d4c
2 parents 547bfc4 + c0ed527
commit d706d4c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/TTS/.models.json b/TTS/.models.json
@@ -76,12 +76,12 @@
  }
  }
  },
- "zh":{
+ "zh-CN":{
  "baker":{
  "tacotron2-DDC-GST":{
- "model_file": "1RR9rZdV_FMm8yvtCHALtUbJf1nxbUiAw",
- "config_file": "1daY1JHGXEozJ-MGYLiWEUmzEwEvM5xpz",
- "stats_file": "1vl9c-D3dW_E7pdhNpDFQLX-giJc0jOtV",
+ "model_file": "1SYpv7V__QYDjKXa_vJmNXo1CSkcoZovy",
+ "config_file": "14BIvfJXnFHi3jcxYNX40__TR6RwJOZqi",
+ "stats_file": "1ECRlXybT6rAWp269CkhjUPwcZ10CkcqD",
  "commit": ""
  }
  }

diff --git a/TTS/tts/utils/text/__init__.py b/TTS/tts/utils/text/__init__.py
@@ -41,7 +41,7 @@ def text2phone(text, language):
  '''
 
  # TO REVIEW : How to have a good implementation for this?
- if language == "chinese-mandarin":
+ if language == "zh-CN":
  ph = chinese_text_to_phonemes(text)
  return ph
 

diff --git a/notebooks/Chinese_Mandarin_DDC_GST_Tacotron2_TTS_and_MultiBand_MelGAN_Example.ipynb b/notebooks/Chinese_Mandarin_DDC_GST_Tacotron2_TTS_and_MultiBand_MelGAN_Example.ipynb
@@ -64,9 +64,9 @@
  ],
  "source": [
  "! mkdir data/\n",
- "! gdown --id 1RR9rZdV_FMm8yvtCHALtUbJf1nxbUiAw -O data/tts_model.pth.tar\n",
- "! gdown --id 1daY1JHGXEozJ-MGYLiWEUmzEwEvM5xpz -O data/tts_config.json\n",
- "! gdown --id 1vl9c-D3dW_E7pdhNpDFQLX-giJc0jOtV -O data/tts_scale_stats.npy"
+ "! gdown --id 1SYpv7V__QYDjKXa_vJmNXo1CSkcoZovy -O data/tts_model.pth.tar\n",
+ "! gdown --id 14BIvfJXnFHi3jcxYNX40__TR6RwJOZqi -O data/tts_config.json\n",
+ "! gdown --id 1ECRlXybT6rAWp269CkhjUPwcZ10CkcqD -O data/tts_scale_stats.npy"
  ]
  },
  {
@@ -526,4 +526,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}