blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

添加了自定义修改随机数种子，方便复现结果。

Browse Source
This commit is contained in:
chasonjiang 2024-03-15 14:34:10 +08:00
parent b8ce03fd1b
commit a2f2a5f4a7
2 changed files with 25 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
GPT_SoVITS/TTS_infer_pack/TTS.py
View File

@ -51,17 +51,23 @@ custom:
""" """
# def set_seed(seed): def set_seed(seed:int):
# random.seed(seed) seed = int(seed)
# os.environ['PYTHONHASHSEED'] = str(seed) seed = seed if seed != -1 else random.randrange(1 << 32)
# np.random.seed(seed) print(f"Set seed to {seed}")
# torch.manual_seed(seed) random.seed(seed)
# torch.cuda.manual_seed(seed) np.random.seed(seed)
# torch.cuda.manual_seed_all(seed) torch.manual_seed(seed)
# torch.backends.cudnn.deterministic = True try:
# torch.backends.cudnn.benchmark = False if torch.cuda.is_available():
# torch.backends.cudnn.enabled = True torch.cuda.manual_seed(seed)
# set_seed(1234) torch.cuda.manual_seed_all(seed)
# torch.backends.cudnn.deterministic = True
# torch.backends.cudnn.benchmark = False
# torch.backends.cudnn.enabled = True
except:
pass
return seed
class TTS_Config: class TTS_Config:
default_configs={ default_configs={
@ -563,6 +569,7 @@ class TTS:
"return_fragment": False, # bool. step by step return the audio fragment. "return_fragment": False, # bool. step by step return the audio fragment.
"speed_factor":1.0, # float. control the speed of the synthesized audio. "speed_factor":1.0, # float. control the speed of the synthesized audio.
"fragment_interval":0.3, # float. to control the interval of the audio fragment. "fragment_interval":0.3, # float. to control the interval of the audio fragment.
"seed": -1, # int. random seed for reproducibility.
} }
returns: returns:
tulpe[int, np.ndarray]: sampling rate and audio data. tulpe[int, np.ndarray]: sampling rate and audio data.
@ -584,6 +591,9 @@ class TTS:
split_bucket = inputs.get("split_bucket", True) split_bucket = inputs.get("split_bucket", True)
return_fragment = inputs.get("return_fragment", False) return_fragment = inputs.get("return_fragment", False)
fragment_interval = inputs.get("fragment_interval", 0.3) fragment_interval = inputs.get("fragment_interval", 0.3)
seed = inputs.get("seed", -1)
set_seed(seed)
if return_fragment: if return_fragment:
# split_bucket = False # split_bucket = False

4
GPT_SoVITS/inference_webui.py
View File

@ -92,6 +92,7 @@ def inference(text, text_lang,
text_split_method, batch_size, text_split_method, batch_size,
speed_factor, ref_text_free, speed_factor, ref_text_free,
split_bucket,fragment_interval, split_bucket,fragment_interval,
seed,
): ):
inputs={ inputs={
"text": text, "text": text,
@ -108,6 +109,7 @@ def inference(text, text_lang,
"split_bucket":split_bucket, "split_bucket":split_bucket,
"return_fragment":False, "return_fragment":False,
"fragment_interval":fragment_interval, "fragment_interval":fragment_interval,
"seed":seed,
} }
for item in tts_pipline.run(inputs): for item in tts_pipline.run(inputs):
@ -203,6 +205,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
) )
with gr.Row(): with gr.Row():
split_bucket = gr.Checkbox(label=i18n("数据分桶(可能会降低一点计算量，选就对了)"), value=True, interactive=True, show_label=True) split_bucket = gr.Checkbox(label=i18n("数据分桶(可能会降低一点计算量，选就对了)"), value=True, interactive=True, show_label=True)
seed = gr.Number(label=i18n("随机种子"),value=-1)
# with gr.Column(): # with gr.Column():
output = gr.Audio(label=i18n("输出的语音")) output = gr.Audio(label=i18n("输出的语音"))
with gr.Row(): with gr.Row():
@ -219,6 +222,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
how_to_cut, batch_size, how_to_cut, batch_size,
speed_factor, ref_text_free, speed_factor, ref_text_free,
split_bucket,fragment_interval, split_bucket,fragment_interval,
seed
], ],
[output], [output],
) )