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):