Skip to content

Commit fee6ffe

Browse files
authored
fix: fairseq model
1 parent 5dcc16d commit fee6ffe

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

TTS/utils/manage.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -260,10 +260,13 @@ def set_model_url(model_item: Dict):
260260
def _set_model_item(self, model_name):
261261
# fetch model info from the dict
262262
if "fairseq" in model_name:
263-
model_type = "tts_models"
264-
lang = model_name.split("/")[1]
263+
split = model_name.split("/")
264+
model_type = split[0]
265+
lang = split[1]
266+
dataset = split[2]
267+
model = split[3]
265268
model_item = {
266-
"model_type": "tts_models",
269+
"model_type": model_type,
267270
"license": "CC BY-NC 4.0",
268271
"default_vocoder": None,
269272
"author": "fairseq",

0 commit comments

Comments
 (0)