support sovits v3 lora training, 8G GPU memory is enough

support sovits v3 lora training, 8G GPU memory is enough
This commit is contained in:
RVC-Boss 2025-02-23 10:46:14 +08:00 committed by GitHub
parent e6a32e15b0
commit 514fb692db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -117,7 +117,7 @@ def set_default():
# minmem = 14
# except RuntimeError as _:
# print("显存不足以开启V3训练")
default_batch_size = minmem // 2 if version!="v3"else minmem//14
default_batch_size = minmem // 2 if version!="v3"else minmem//8
default_batch_size_s1=minmem // 2
else:
gpu_info = ("%s\t%s" % ("0", "CPU"))