Skip to content

Commit

Permalink
Extract dataset and model for fairseq models
Browse files Browse the repository at this point in the history
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}"
  • Loading branch information
hanasim committed May 29, 2024
1 parent dbf1a08 commit 0d1aa58
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions TTS/utils/manage.py
Original file line number Diff line number Diff line change
Expand Up @@ -262,6 +262,8 @@ def _set_model_item(self, model_name):
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",
Expand Down

0 comments on commit 0d1aa58

Please sign in to comment.