From acc83cd3e66fb6329c203d415636b7493221acba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Sun, 20 Feb 2022 11:51:37 +0100
Subject: [PATCH] Update Vits model API

---
 TTS/tts/models/vits.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
index 0c795ca1..6ff53c71 100644
--- a/TTS/tts/models/vits.py
+++ b/TTS/tts/models/vits.py
@@ -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