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 00:37:14 +08:00
committed by GitHub
parent 56509a17c9
commit e937b625e4
4 changed files with 488 additions and 55 deletions

View File

@@ -26,12 +26,7 @@ from AR.utils import get_newest_ckpt
from collections import OrderedDict
from time import time as ttime
import shutil
def my_save(fea,path):#####fix issue: torch.save doesn't support chinese path
dir=os.path.dirname(path)
name=os.path.basename(path)
tmp_path="%s.pth"%(ttime())
torch.save(fea,tmp_path)
shutil.move(tmp_path,"%s/%s"%(dir,name))
from process_ckpt import my_save
class my_model_ckpt(ModelCheckpoint):