Spaces:
Runtime error
Runtime error
ArissBandoss
commited on
Update goai_helpers/goai_tts2.py
Browse files
goai_helpers/goai_tts2.py
CHANGED
@@ -73,7 +73,7 @@ class MooreTTS:
|
|
73 |
# charger le point de contrôle du modèle dans le modèle initialisé
|
74 |
self.model.load_checkpoint(
|
75 |
self.config,
|
76 |
-
checkpoint_path=self.local_dir+ "/
|
77 |
vocab_path=self.paths['vocab.json'],
|
78 |
use_deepspeed=False
|
79 |
)
|
@@ -134,7 +134,7 @@ class MooreTTS:
|
|
134 |
Returns :
|
135 |
Un dictionnaire avec des clés comme noms des composants et des valeurs comme chemins des fichiers.
|
136 |
"""
|
137 |
-
components = ['
|
138 |
return {name: os.path.join(local_dir, name) for name in components}
|
139 |
|
140 |
def text_to_speech(
|
|
|
73 |
# charger le point de contrôle du modèle dans le modèle initialisé
|
74 |
self.model.load_checkpoint(
|
75 |
self.config,
|
76 |
+
checkpoint_path=self.local_dir+ "/model_compressed.pth" ,
|
77 |
vocab_path=self.paths['vocab.json'],
|
78 |
use_deepspeed=False
|
79 |
)
|
|
|
134 |
Returns :
|
135 |
Un dictionnaire avec des clés comme noms des composants et des valeurs comme chemins des fichiers.
|
136 |
"""
|
137 |
+
components = ['model_compressed.pth', 'config.json', 'vocab.json', 'dvae.pth', 'mel_stats.pth']
|
138 |
return {name: os.path.join(local_dir, name) for name in components}
|
139 |
|
140 |
def text_to_speech(
|