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