Update train_vits-2.py

train_vits-2.py

@@ -10,7 +10,7 @@ from TTS.tts.models.vits import  CharactersConfig, Vits, VitsArgs, VitsAudioConf
 from TTS.tts.utils.text.tokenizer import TTSTokenizer
 from TTS.utils.audio import AudioProcessor
 from TTS.tts.utils.speakers import SpeakerManager
-
+from TTS.tts.datasets.formatters import mozilla_with_speaker
 
 output_path = os.path.dirname(os.path.abspath(__file__))
 
@@ -134,6 +134,7 @@ tokenizer, config = TTSTokenizer.init_from_config(config)
 # Load all the datasets samples and split traning and evaluation sets
 train_samples, eval_samples = load_tts_samples(
     config.datasets,
+    formatter=mozilla_with_speaker,
     eval_split=True,
     eval_split_max_size=config.eval_split_max_size,
     eval_split_size=config.eval_split_size,