feat: melotts init time

This commit is contained in:
superobk
2024-05-23 15:51:35 +08:00
parent d149521b11
commit 3843034c7b

View File

@ -27,8 +27,8 @@ class MeloTTS(Blackbox):
language: str
speaker: str
@inject
def __init__(self, melo_config: MeloConf) -> None:
@logging_time(logger=logger)
def model_init(self, melo_config: MeloConf) -> None:
self.speed = melo_config.speed
self.device = melo_config.device
self.language = melo_config.language
@ -45,6 +45,10 @@ class MeloTTS(Blackbox):
self.url = melo_config.url
logging.info('#### Initializing MeloTTS Service in ' + self.device + ' mode...')
@inject
def __init__(self, melo_config: MeloConf) -> None:
self.model_init(melo_config)
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)