From 402346e4932743797e7ab3f0fbf53789c6787e66 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Sat, 22 Jun 2024 15:18:57 +0900 Subject: [PATCH] Fix mid_result assignment --- gradio_tabs/inference.py | 2 +- style_bert_vits2/tts_model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/gradio_tabs/inference.py b/gradio_tabs/inference.py index 2673151..3f93a15 100644 --- a/gradio_tabs/inference.py +++ b/gradio_tabs/inference.py @@ -223,7 +223,7 @@ def change_null_model_row( mid_result = {} mid_result["name"] = null_model_name mid_result["path"] = null_model_path - mid_result["weight"] = null_tempo_weights + mid_result["weight"] = null_voice_weights mid_result["pitch"] = null_voice_pitch_weights mid_result["style"] = null_speech_style_weights mid_result["tempo"] = null_tempo_weights diff --git a/style_bert_vits2/tts_model.py b/style_bert_vits2/tts_model.py index 96c68cc..2ad6f4b 100644 --- a/style_bert_vits2/tts_model.py +++ b/style_bert_vits2/tts_model.py @@ -297,7 +297,7 @@ class TTSModel: given_tone (Optional[list[int]], optional): アクセントのトーンのリスト. Defaults to None. pitch_scale (float, optional): ピッチの高さ (1.0 から変更すると若干音質が低下する). Defaults to 1.0. intonation_scale (float, optional): 抑揚の平均からの変化幅 (1.0 から変更すると若干音質が低下する). Defaults to 1.0. - null_model_params(dict[int,dict[str,Union[str,float]],optional):推論時に使用するヌルモデルの名前、適用割合のdictが入ったdict。 + null_model_params (dict[int, dict[str, Union[str, float]]], optional): 推論時に使用するヌルモデルの名前、適用割合のdictが入ったdict。 Returns: tuple[int, NDArray[Any]]: サンプリングレートと音声データ (16bit PCM) """