from TTS.utils.generic_utils import find_module def setup_model(config: "Coqpit") -> "BaseTTS": print(" > Using model: {}".format(config.model)) # fetch the right model implementation. if "base_model" in config and config["base_model"] is not None: MyModel = find_module("TTS.tts.models", config.base_model.lower()) else: MyModel = find_module("TTS.tts.models", config.model.lower()) model = MyModel.init_from_config(config) return model