mirror of https://github.com/coqui-ai/TTS.git
Update base_vocoder
parent
fc3b6d2861
commit
833de62e30
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue