Extract dataset and model for fairseq models

When we use the model "tts_models/eng/fairseq/vits", got the following error.

Error in text to speech: cannot access local variable 'dataset' where it is not associated with a value

at line ..../.conda/envs/venv/lib/python3.11/site-packages/TTS/utils/manage.py", line 304, in _set_model_item
    model_full_name = f"{model_type}--{lang}--{dataset}--{model}"
This commit is contained in:
Manju 2024-05-29 21:46:42 +01:00 committed by GitHub
parent dbf1a08a0d
commit 0d1aa58fca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions

View File

@ -262,6 +262,8 @@ class ModelManager(object):
if "fairseq" in model_name:
model_type = "tts_models"
lang = model_name.split("/")[1]
dataset = model_name.split("/")[2]
model = model_name.split("/")[3]
model_item = {
"model_type": "tts_models",
"license": "CC BY-NC 4.0",