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. """ ... diff --git a/hubconf.py b/hubconf.py index 0c9c5930fc..64a1ce27b9 100644 --- a/hubconf.py +++ b/hubconf.py @@ -10,7 +10,7 @@ def tts(model_name='tts_models/en/ljspeech/tacotron2-DCA', vocoder_name=None, use_cuda=False): - """TTS entry point for PyTorch Hub that provides a Synthesizer object to synthesize speech from a give text. + """TTS entry point for PyTorch Hub that provides a Synthesizer object to synthesize speech from a given text. Example: >>> synthesizer = torch.hub.load('coqui-ai/TTS', 'tts', source='github')