From 55568eaf7ef1cf580a674a8a3eca1805b88912d4 Mon Sep 17 00:00:00 2001 From: Vignesh Skanda Date: Mon, 30 Sep 2024 11:58:01 +0530 Subject: [PATCH] Update model.py --- TTS/model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/TTS/model.py b/TTS/model.py index ae6be7b444..b7fce033fa 100644 --- a/TTS/model.py +++ b/TTS/model.py @@ -28,7 +28,7 @@ def inference(self, input: torch.Tensor, aux_input={}) -> Dict: """Forward pass for inference. It must return a dictionary with the main model output and all the auxiliary outputs. The key ```model_outputs``` - is considered to be the main output and you can add any other auxiliary outputs as you want. + is considered to be the main output and you can add any other auxiliary outputs if you want. We don't use `*kwargs` since it is problematic with the TorchScript API. @@ -54,6 +54,7 @@ def load_checkpoint( checkpoint_path (str): Path to the model checkpoint file. eval (bool, optional): If true, init model for inference else for training. Defaults to False. strict (bool, optional): Match all checkpoint keys to model's keys. Defaults to True. - cache (bool, optional): If True, cache the file locally for subsequent calls. It is cached under `get_user_data_dir()/tts_cache`. Defaults to False. + cache (bool, optional): If True, cache the file locally for subsequent calls. It is cached under `get_user_data_dir()/tts_cache`. + Defaults to False. """ ...