Update base_vocoder

pull/1324/head
Eren Gölge 2022-02-20 11:38:24 +01:00
parent fc3b6d2861
commit 833de62e30
1 changed files with 3 additions and 3 deletions

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