ArissBandoss commited on
Commit
0d77897
·
verified ·
1 Parent(s): e44f8d6

Update goai_helpers/goai_tts2.py

Browse files
Files changed (1) hide show
  1. goai_helpers/goai_tts2.py +2 -2
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+ "/best_model_28574.pth" ,
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 = ['best_model_28574.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(
 
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(