From afbcf4007a8645ea187ef426c0c95aa34d2d7d84 Mon Sep 17 00:00:00 2001
From: bstr9 <82707826+bstr9@users.noreply.github.com>
Date: Wed, 26 Jun 2024 22:26:20 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8Dprompt=5Fcache?=
 =?UTF-8?q?=E9=87=8C=E9=9D=A2=E6=B2=A1=E6=9C=89cache=E4=BD=8Fref=5Faudio?=
 =?UTF-8?q?=5Fpath=EF=BC=8C=E5=AF=BC=E8=87=B4=E6=AF=8F=E6=AC=A1TTS?=
 =?UTF-8?q?=E6=8E=A8=E7=90=86=E9=83=BD=E9=9C=80=E8=A6=81=E5=8A=A0=E8=BD=BD?=
 =?UTF-8?q?=E4=B8=80=E9=81=8Dreference=5Faudio=E7=9A=84bug=E3=80=90?=
 =?UTF-8?q?=E9=A6=96=E6=AC=A1=E9=9F=B3=E9=A2=91=E5=B8=A7=E4=BB=8E3.2s=20->?=
 =?UTF-8?q?=201.7s=E3=80=91=20(#1235)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 GPT_SoVITS/TTS_infer_pack/TTS.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/GPT_SoVITS/TTS_infer_pack/TTS.py b/GPT_SoVITS/TTS_infer_pack/TTS.py
index eaacb52..e36edeb 100644
--- a/GPT_SoVITS/TTS_infer_pack/TTS.py
+++ b/GPT_SoVITS/TTS_infer_pack/TTS.py
@@ -360,7 +360,11 @@ class TTS:
         '''
         self._set_prompt_semantic(ref_audio_path)
         self._set_ref_spec(ref_audio_path)
+        self._set_ref_audio_path(ref_audio_path)
         
+    def _set_ref_audio_path(self, ref_audio_path):
+        self.prompt_cache["ref_audio_path"] = ref_audio_path 
+
     def _set_ref_spec(self, ref_audio_path):
         audio = load_audio(ref_audio_path, int(self.configs.sampling_rate))
         audio = torch.FloatTensor(audio)