bacow/GPT-SoVITS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Code refactor + remove unused imports

Browse Source
This commit is contained in:
Blaise
2024-01-16 17:10:27 +01:00
parent 9031ac9a92
commit 0d92575115
5 changed files with 671 additions and 335 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
GPT_SoVITS/process_ckpt.py
View File

@@ -1,11 +1,12 @@
import os
import sys
import traceback
from collections import OrderedDict
import torch
from tools.i18n.i18n import I18nAuto
i18n = I18nAuto()
def savee(ckpt, name, epoch, steps, hps):
try:
opt = OrderedDict()
@@ -15,8 +16,8 @@ def savee(ckpt, name, epoch, steps, hps):
continue
opt["weight"][key] = ckpt[key].half()
opt["config"] = hps
opt["info"] = "%sepoch_%siteration" % (epoch,steps)
torch.save(opt, "%s/%s.pth" % (hps.save_weight_dir,name))
opt["info"] = "%sepoch_%siteration" % (epoch, steps)
torch.save(opt, "%s/%s.pth" % (hps.save_weight_dir, name))
return "Success."
except:
return traceback.format_exc()