修复了一些bug (#994)

2024-04-21 20:28:33 +08:00
parent 29f22115fb
commit 35e755427d
2 changed files with 2 additions and 2 deletions
--- a/GPT_SoVITS/TTS_infer_pack/TTS.py
+++ b/GPT_SoVITS/TTS_infer_pack/TTS.py
@@ -749,7 +749,7 @@ class TTS:
                if no_prompt_text :
                    prompt = None
                else:
-                    prompt = self.prompt_cache["prompt_semantic"].expand(all_phoneme_ids.shape[0], -1).to(self.configs.device)
+                    prompt = self.prompt_cache["prompt_semantic"].expand(len(all_phoneme_ids), -1).to(self.configs.device)


                pred_semantic_list, idx_list = self.t2s_model.model.infer_panel(