From 833de62e3059fdc76bbafd5afeb2d34793ac8736 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Sun, 20 Feb 2022 11:38:24 +0100
Subject: [PATCH] Update base_vocoder

---
 TTS/vocoder/models/base_vocoder.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/TTS/vocoder/models/base_vocoder.py b/TTS/vocoder/models/base_vocoder.py
index 2728525c..01a7ff68 100644
--- a/TTS/vocoder/models/base_vocoder.py
+++ b/TTS/vocoder/models/base_vocoder.py
@@ -1,11 +1,11 @@
 from coqpit import Coqpit
 
-from TTS.model import BaseModel
+from TTS.model import BaseTrainerModel
 
 # pylint: skip-file
 
 
-class BaseVocoder(BaseModel):
+class BaseVocoder(BaseTrainerModel):
     """Base `vocoder` class. Every new `vocoder` model must inherit this.
 
     It defines `vocoder` specific functions on top of `Model`.
@@ -19,7 +19,7 @@ class BaseVocoder(BaseModel):
     """
 
     def __init__(self, config):
-        super().__init__(config)
+        super().__init__()
         self._set_model_args(config)
 
     def _set_model_args(self, config: Coqpit):