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 Vits model API

This commit is contained in:
Eren Gölge 2022-02-20 11:51:37 +01:00
parent fe656659be
commit acc83cd3e6
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
TTS/tts/models/vits.py
View File

@ -1,8 +1,9 @@
import collections
import math
import os
from dataclasses import dataclass, field, replace
from itertools import chain
from typing import Dict, List, Tuple, Union
from typing import Dict, List, Optional, Tuple, Union
import torch
import torch.distributed as dist
@ -544,8 +545,7 @@ class Vits(BaseTTS):
ap: "AudioProcessor" = None,
tokenizer: "TTSTokenizer" = None,
speaker_manager: SpeakerManager = None,
language_manager: LanguageManager = None,
):
language_manager: LanguageManager = None,):
super().__init__(config, ap, tokenizer, speaker_manager, language_manager)
@ -1483,6 +1483,10 @@ class Vits(BaseTTS):
language_manager = LanguageManager.init_from_config(config)
return Vits(new_config, ap, tokenizer, speaker_manager, language_manager)
##################################
# VITS CHARACTERS
##################################
##################################
# VITS CHARACTERS