Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request from idiap/fix-language-manager 

fix(LanguageManager): allow initialisation from config with language ids file
This commit is contained in:
Enno Hermann 2024-04-23 13:50:48 +02:00 committed by GitHub
parent 5527f70d68 52a52b5e21
commit d65ba4ad6a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
TTS/tts/utils
languages.py

12
TTS/tts/utils/languages.py
View File

@ -1,5 +1,5 @@
import os import os
from typing import Any, Dict, List from typing import Any, Dict, List, Optional
import fsspec import fsspec
import numpy as np import numpy as np
@ -85,18 +85,18 @@ class LanguageManager(BaseIDManager):
self._save_json(file_path, self.name_to_id) self._save_json(file_path, self.name_to_id)
@staticmethod @staticmethod
def init_from_config(config: Coqpit) -> "LanguageManager": def init_from_config(config: Coqpit) -> Optional["LanguageManager"]:
"""Initialize the language manager from a Coqpit config. """Initialize the language manager from a Coqpit config.
Args: Args:
config (Coqpit): Coqpit config. config (Coqpit): Coqpit config.
""" """
language_manager = None
if check_config_and_model_args(config, "use_language_embedding", True): if check_config_and_model_args(config, "use_language_embedding", True):
if config.get("language_ids_file", None): if config.get("language_ids_file", None):
language_manager = LanguageManager(language_ids_file_path=config.language_ids_file) return LanguageManager(language_ids_file_path=config.language_ids_file)
language_manager = LanguageManager(config=config) # Fall back to parse language IDs from the config
return language_manager return LanguageManager(config=config)
return None
def _set_file_path(path): def _set_file_path(path):