mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-04-06 03:57:44 +08:00
23 lines
639 B
Python
23 lines
639 B
Python
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()
|
|
opt["weight"] = {}
|
|
for key in ckpt.keys():
|
|
if "enc_q" in key:
|
|
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))
|
|
return "Success."
|
|
except:
|
|
return traceback.format_exc()
|