Update base_vocoder

This commit is contained in:
Eren Gölge
2022-02-25 11:28:14 +01:00
parent fc3b6d2861
commit 833de62e30
+3 -3
View File
@@ -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):