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

CPU 使用更低 batch size (#2067)

Browse Source 
* Update webui.py

lower batch size on cpu

* Update webui.py

Improvements
This commit is contained in:
Lion-Wu 2025-02-17 16:42:35 +08:00 committed by GitHub
parent 087fd24579
commit c70daefea2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
webui.py
View File

@ -103,7 +103,7 @@ def set_default():
gpu_info = ("%s\t%s" % ("0", "CPU")) gpu_info = ("%s\t%s" % ("0", "CPU"))
gpu_infos.append("%s\t%s" % ("0", "CPU")) gpu_infos.append("%s\t%s" % ("0", "CPU"))
set_gpu_numbers.add(0) set_gpu_numbers.add(0)
default_batch_size = default_batch_size_s1=int(psutil.virtual_memory().total/ 1024 / 1024 / 1024 / 2) default_batch_size = default_batch_size_s1 = int(psutil.virtual_memory().total/ 1024 / 1024 / 1024 / 4)
if version!="v3": if version!="v3":
default_sovits_epoch=8 default_sovits_epoch=8
default_sovits_save_every_epoch=4 default_sovits_save_every_epoch=4
@ -114,6 +114,9 @@ def set_default():
default_sovits_save_every_epoch=1 default_sovits_save_every_epoch=1
max_sovits_epoch=3 max_sovits_epoch=3
max_sovits_save_every_epoch=3 max_sovits_save_every_epoch=3
default_batch_size = max(1, default_batch_size)
default_batch_size_s1 = max(1, default_batch_size_s1)
default_max_batch_size=default_batch_size*3 default_max_batch_size=default_batch_size*3
set_default() set_default()