mirror of https://github.com/coqui-ai/TTS.git
13 lines
480 B
Python
13 lines
480 B
Python
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
|