blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-12-18 19:07:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix training error caused by float type of default_batch_size parameter

Browse Source
This commit is contained in:
Spr_Aachen 2025-11-10 07:02:53 +08:00
parent 11aa78bd9b
commit c5d57d7afa
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
webui.py
View File

@ -117,8 +117,8 @@ def set_default():
gpu_info = "\n".join(gpu_infos) gpu_info = "\n".join(gpu_infos)
if is_gpu_ok: if is_gpu_ok:
minmem = min(mem) minmem = min(mem)
default_batch_size = minmem // 2 if version not in v3v4set else minmem // 8 default_batch_size = int(minmem // 2 if version not in v3v4set else minmem // 8)
default_batch_size_s1 = minmem // 2 default_batch_size_s1 = int(minmem // 2)
else: else:
default_batch_size = default_batch_size_s1 = int(psutil.virtual_memory().total / 1024 / 1024 / 1024 / 4) default_batch_size = default_batch_size_s1 = int(psutil.virtual_memory().total / 1024 / 1024 / 1024 / 4)
if version not in v3v4set: if version not in v3v4set: