wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update Vits model API

Browse Source
This commit is contained in:
Eren Gölge
2022-02-25 11:31:56 +01:00
parent fe656659be
commit acc83cd3e6
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/models/vits.py
+7 -3
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