Merge commit '6e2b49186c5b961f0de41ea485d398dffa9787b4' into Doc

This commit is contained in:
starylan 2025-02-12 01:49:09 +08:00
commit cb4e31e927
2 changed files with 3 additions and 1 deletions

View File

@ -34,3 +34,4 @@ opencc; sys_platform != 'linux'
opencc==1.1.1; sys_platform == 'linux'
python_mecab_ko; sys_platform != 'win32'
fastapi<0.112.2
x_transformers

View File

@ -90,11 +90,12 @@ if torch.cuda.is_available() or ngpu != 0:
# gpu_infos.append("%s\t%s" % ("0", "Apple GPU"))
# mem.append(psutil.virtual_memory().total/ 1024 / 1024 / 1024) # 实测使用系统内存作为显存不会爆显存
minmem=min(mem)
def set_default():
global default_batch_size,default_max_batch_size,gpu_info,default_sovits_epoch,default_sovits_save_every_epoch,max_sovits_epoch,max_sovits_save_every_epoch,default_batch_size_s1
if if_gpu_ok and len(gpu_infos) > 0:
gpu_info = "\n".join(gpu_infos)
minmem = min(mem)
default_batch_size = minmem // 2 if version!="v3"else minmem//14
default_batch_size_s1=minmem // 2
else: