APPLY THE FIX OF SAOTOME MODELC

This commit is contained in:
samiabat 2025-05-27 23:49:22 +03:00
parent 15f71b4d65
commit ecdd1eb2ec

20
api.py
View File

@ -1097,11 +1097,18 @@ async def version_4(
GPT_model_path = "GPT_SoVITS/pretrained_models/kurari-e40.ckpt", GPT_model_path = "GPT_SoVITS/pretrained_models/kurari-e40.ckpt",
SoVITS_model_path = "GPT_SoVITS/pretrained_models/kurari_e20_s1800_l32.pth", SoVITS_model_path = "GPT_SoVITS/pretrained_models/kurari_e20_s1800_l32.pth",
ref_text: str = "おはよう〜。今日はどんな1日過ごすーくらりはね〜いつでもあなたの味方だよ", ref_text: str = "おはよう〜。今日はどんな1日過ごすーくらりはね〜いつでもあなたの味方だよ",
ref_language: str = "ja", ref_language: str = "日文",
target_text: str = None, target_text: str = None,
text_language: str = "ja", text_language: str = "日文",
output_path: str = None output_path: str = None,
character_name: str = "Kurari",
): ):
if character_name == "saotome":
GPT_model_path = "GPT_SoVITS/pretrained_models/saotome-e30.ckpt"
SoVITS_model_path = "GPT_SoVITS/pretrained_models/saotome_e20_s1800_l32.pth"
ref_text = "今日は友達と一緒に映画を見に行く予定ですが、天気が悪くて少し心配です。"
ref_language = "日文"
# Create a temporary buffer to store the audio # Create a temporary buffer to store the audio
audio_buffer = io.BytesIO() audio_buffer = io.BytesIO()
# GPT_model_path, SoVITS_model_path, ref_audio_path, ref_text, ref_language, target_text, text_language, output_path # GPT_model_path, SoVITS_model_path, ref_audio_path, ref_text, ref_language, target_text, text_language, output_path
@ -1148,16 +1155,13 @@ async def tts_endpoint(
top_p: float = 1.0, top_p: float = 1.0,
temperature: float = 1.0, temperature: float = 1.0,
speed: float = 1.0, speed: float = 1.0,
sample_steps: int = 32, sample_steps: int = 20,
if_sr: bool = False if_sr: bool = False
): ):
if character == "kurari": if character == "kurari":
prompt_text = "おはよう〜。今日はどんな1日過ごすーくらりはね〜いつでもあなたの味方だよ" prompt_text = "おはよう〜。今日はどんな1日過ごすーくらりはね〜いつでもあなたの味方だよ"
if text_language == "en" and character == "saotome":
refer_wav_path = f"idols/{character}_eng/{character}.wav"
inp_refs = [f"idols/{character}_eng/refs/{file}" for file in os.listdir(f"idols/{character}_eng/refs") if file.endswith('.wav')]
else:
refer_wav_path = f"idols/{character}/{character}.wav" refer_wav_path = f"idols/{character}/{character}.wav"
inp_refs = [f"idols/{character}/refs/{file}" for file in os.listdir(f"idols/{character}/refs") if file.endswith('.wav')] inp_refs = [f"idols/{character}/refs/{file}" for file in os.listdir(f"idols/{character}/refs") if file.endswith('.wav')]