From 7be9056b3dc33c14d2c122986bdb474fc268f353 Mon Sep 17 00:00:00 2001
From: Edresson Casanova <edresson1@gmail.com>
Date: Thu, 31 Mar 2022 11:05:58 -0300
Subject: [PATCH] Remove useless encoder weights reload

---
 TTS/tts/models/vits.py       | 5 +----
 tests/tts_tests/test_vits.py | 2 ++
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
index 596d28c2..ecd43dba 100644
--- a/TTS/tts/models/vits.py
+++ b/TTS/tts/models/vits.py
@@ -677,8 +677,6 @@ class Vits(BaseTTS):
                     " [!] To use the speaker consistency loss (SCL) you need to specify encoder_model_path and encoder_config_path !!"
                 )
 
-            # load encoder
-            self.speaker_manager.init_encoder(self.args.encoder_model_path, self.args.encoder_config_path)
             self.speaker_manager.encoder.eval()
             print(" > External Speaker Encoder Loaded !!")
 
@@ -699,8 +697,7 @@ class Vits(BaseTTS):
                 raise RuntimeError(
                     " [!] To use the emotion consistency loss (ECL) you need to specify encoder_model_path and encoder_config_path !!"
                 )
-            # load encoder
-            self.emotion_manager.init_encoder(self.args.encoder_model_path, self.args.encoder_config_path)
+
             self.emotion_manager.encoder.eval()
             print(" > External Emotion Encoder Loaded !!")
 
diff --git a/tests/tts_tests/test_vits.py b/tests/tts_tests/test_vits.py
index cc8c7b5c..97816dbd 100644
--- a/tests/tts_tests/test_vits.py
+++ b/tests/tts_tests/test_vits.py
@@ -386,6 +386,8 @@ class TestVits(unittest.TestCase):
         batch["d_vectors"] = None
         batch["speaker_ids"] = None
         batch["language_ids"] = None
+        batch["emotion_embeddings"] = None
+        batch["emotion_ids"] = None
         return batch
 
     def test_train_step(self):