Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update model.py

This commit is contained in:
Vignesh Skanda 2024-09-30 11:58:01 +05:30 committed by GitHub
parent cb5d9bad41
commit 55568eaf7e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/model.py
View File

@ -28,7 +28,7 @@ class BaseTrainerModel(TrainerModel):
"""Forward pass for inference. """Forward pass for inference.
It must return a dictionary with the main model output and all the auxiliary outputs. The key ```model_outputs``` 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. We don't use `*kwargs` since it is problematic with the TorchScript API.
@ -54,6 +54,7 @@ class BaseTrainerModel(TrainerModel):
checkpoint_path (str): Path to the model checkpoint file. checkpoint_path (str): Path to the model checkpoint file.
eval (bool, optional): If true, init model for inference else for training. Defaults to False. 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. 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.
""" """
... ...