mirror of https://github.com/coqui-ai/TTS.git
Update base_vocoder
This commit is contained in:
parent
fc3b6d2861
commit
833de62e30
|
@ -1,11 +1,11 @@
|
||||||
from coqpit import Coqpit
|
from coqpit import Coqpit
|
||||||
|
|
||||||
from TTS.model import BaseModel
|
from TTS.model import BaseTrainerModel
|
||||||
|
|
||||||
# pylint: skip-file
|
# pylint: skip-file
|
||||||
|
|
||||||
|
|
||||||
class BaseVocoder(BaseModel):
|
class BaseVocoder(BaseTrainerModel):
|
||||||
"""Base `vocoder` class. Every new `vocoder` model must inherit this.
|
"""Base `vocoder` class. Every new `vocoder` model must inherit this.
|
||||||
|
|
||||||
It defines `vocoder` specific functions on top of `Model`.
|
It defines `vocoder` specific functions on top of `Model`.
|
||||||
|
@ -19,7 +19,7 @@ class BaseVocoder(BaseModel):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
super().__init__(config)
|
super().__init__()
|
||||||
self._set_model_args(config)
|
self._set_model_args(config)
|
||||||
|
|
||||||
def _set_model_args(self, config: Coqpit):
|
def _set_model_args(self, config: Coqpit):
|
||||||
|
|
Loading…
Reference in New Issue