From 0d1aa58fcaa6cd28e0a04581d723068dfa2edaa2 Mon Sep 17 00:00:00 2001
From: Manju <1180369+hanasim@users.noreply.github.com>
Date: Wed, 29 May 2024 21:46:42 +0100
Subject: [PATCH] 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}"
---
 TTS/utils/manage.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py
index 3a527f46..35e8ae2a 100644
--- a/TTS/utils/manage.py
+++ b/TTS/utils/manage.py
@@ -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",