From 4c5cb44eeb528db360e8a0837b16597fa334c4c4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Tue, 7 Dec 2021 12:56:44 +0000 Subject: [PATCH] Update setup_model --- TTS/tts/models/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/TTS/tts/models/__init__.py b/TTS/tts/models/__init__.py index cb1c2e21..d76a3beb 100644 --- a/TTS/tts/models/__init__.py +++ b/TTS/tts/models/__init__.py @@ -1,12 +1,14 @@ +from typing import Dict, List, Union + from TTS.utils.generic_utils import find_module -def setup_model(config: "Coqpit") -> "BaseTTS": +def setup_model(config: "Coqpit", samples: Union[List[List], List[Dict]] = None) -> "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) + model = MyModel.init_from_config(config, samples) return model