feat: melotts init time

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

View File

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