Krisshvamsi
commited on
Commit
•
c6f3d36
1
Parent(s):
cee8eee
Update TTSModel.py
Browse files- TTSModel.py +1 -1
TTSModel.py
CHANGED
@@ -28,7 +28,7 @@ class TTSModel(Pretrained):
|
|
28 |
def __init__(self, *args, **kwargs):
|
29 |
super().__init__(*args, **kwargs)
|
30 |
self.label_encoder = self.hparams.label_encoder
|
31 |
-
|
32 |
self.g2p = GraphemeToPhoneme.from_hparams("speechbrain/soundchoice-g2p")
|
33 |
|
34 |
|
|
|
28 |
def __init__(self, *args, **kwargs):
|
29 |
super().__init__(*args, **kwargs)
|
30 |
self.label_encoder = self.hparams.label_encoder
|
31 |
+
self.label_encoder.update_from_iterable(self.hparams["lexicon"], sequence_input=False)
|
32 |
self.g2p = GraphemeToPhoneme.from_hparams("speechbrain/soundchoice-g2p")
|
33 |
|
34 |
|