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

修复sovits版本判断

Browse Source
This commit is contained in:
RVC-Boss 2025-02-12 14:48:36 +08:00 committed by GitHub
parent 598cddabaf
commit 8e9e3c07d5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
GPT_SoVITS/inference_webui.py
View File

@ -197,21 +197,32 @@ def change_sovits_weights(sovits_path,prompt_language=None,text_language=None):
v1:about 82942KB v1:about 82942KB
half thr:82978KB half thr:82978KB
v2:about 83014KB v2:about 83014KB
half thr:100MB
v1base:103490KB
half thr:103520KB
v2base:103551KB
v3:about 750MB v3:about 750MB
~82978K~100M~103420~700M
v1-v2-v1base-v2base-v3
version: version:
symbols version and timebre_embedding version symbols version and timebre_embedding version
model_version: model_version:
sovits is v1/2 (VITS) or v3 (shortcut CFM DiT) sovits is v1/2 (VITS) or v3 (shortcut CFM DiT)
''' '''
if os.path.getsize(sovits_path)>82978*1024: size=os.path.getsize(sovits_path)
if size<82978*1024:
model_version=version="v1"
elif size<100*1024*1024:
model_version=version="v2"
elif size<103520*1024:
model_version=version="v1"
elif size<700*1024*1024:
model_version = version = "v2"
else:
version = "v2" version = "v2"
else:
version="v1"
if os.path.getsize(sovits_path)>700*1024*1024:
model_version="v3" model_version="v3"
else:
model_version=version
if prompt_language is not None and text_language is not None: if prompt_language is not None and text_language is not None:
if prompt_language in list(dict_language.keys()): if prompt_language in list(dict_language.keys()):
prompt_text_update, prompt_language_update = {'__type__':'update'}, {'__type__':'update', 'value':prompt_language} prompt_text_update, prompt_language_update = {'__type__':'update'}, {'__type__':'update', 'value':prompt_language}
@ -248,6 +259,7 @@ def change_sovits_weights(sovits_path,prompt_language=None,text_language=None):
n_speakers=hps.data.n_speakers, n_speakers=hps.data.n_speakers,
**hps.model **hps.model
) )
model_version=version
else: else:
vq_model = SynthesizerTrnV3( vq_model = SynthesizerTrnV3(
hps.data.filter_length // 2 + 1, hps.data.filter_length // 2 + 1,