test

stitched_model.py

@@ -23,13 +23,17 @@ class CombinedModel(nn.Module):
 
         if huggingface_checkpoint:
             self.tts_checkpoint_path = hf_hub_download(huggingface_checkpoint,tts_checkpoint_name)
+            vocab_file = hf_hub_download(huggingface_checkpoint,"vocab.txt")
+            config_file = hf_hub_download(huggingface_checkpoint,"config.json")
         elif not tts_checkpoint_path:
             self.tts_checkpoint_path = self.download_mms_tts(self.language)
+            vocab_file = f"{self.tts_checkpoint_path}/vocab.txt"
+            config_file = f"{self.tts_checkpoint_path}/config.json"
         else:
             self.tts_checkpoint_path = tts_checkpoint_path
-
-        vocab_file = f"{self.tts_checkpoint_path}/vocab.txt"
-        config_file = f"{self.tts_checkpoint_path}/config.json"
+            vocab_file = f"{self.tts_checkpoint_path}/vocab.txt"
+            config_file = f"{self.tts_checkpoint_path}/config.json"
+        
         
         self.hps = utils.get_hparams_from_file(config_file)
         self.text_mapper = TextMapper(vocab_file)