RVC/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-01-20 02:51:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
RVC-Boss
2023-05-14 15:06:50 +08:00
committed by GitHub
parent 3b5a2298d7
commit 404ce9338f
3 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
train/process_ckpt.py
View File

@@ -2,7 +2,7 @@ import torch, traceback, os, pdb
from collections import OrderedDict
def savee(ckpt, sr, if_f0, name, epoch):
def savee(ckpt, sr, if_f0, name, epoch,version):
try:
opt = OrderedDict()
opt["weight"] = {}
@@ -76,6 +76,7 @@ def savee(ckpt, sr, if_f0, name, epoch):
opt["info"] = "%sepoch" % epoch
opt["sr"] = sr
opt["f0"] = if_f0
opt["version"] = version
torch.save(opt, "weights/%s.pth" % name)
return "Success."
except:
@@ -85,16 +86,17 @@ def savee(ckpt, sr, if_f0, name, epoch):
def show_info(path):
try:
a = torch.load(path, map_location="cpu")
return "模型信息:%s\n采样率:%s\n模型是否输入音高引导:%s" % (
return "模型信息:%s\n采样率:%s\n模型是否输入音高引导:%s\n版本:%s" % (
a.get("info", "None"),
a.get("sr", "None"),
a.get("f0", "None"),
a.get("version", "None"),
)
except:
return traceback.format_exc()
def extract_small_model(path, name, sr, if_f0, info):
def extract_small_model(path, name, sr, if_f0, info,version):
try:
ckpt = torch.load(path, map_location="cpu")
if "model" in ckpt:
@@ -171,6 +173,7 @@ def extract_small_model(path, name, sr, if_f0, info):
if info == "":
info = "Extracted model."
opt["info"] = info
opt["version"] = version
opt["sr"] = sr
opt["f0"] = int(if_f0)
torch.save(opt, "weights/%s.pth" % name)
@@ -191,7 +194,7 @@ def change_info(path, info, name):
return traceback.format_exc()
def merge(path1, path2, alpha1, sr, f0, info, name):
def merge(path1, path2, alpha1, sr, f0, info, name,version):
try:
def extract(ckpt):
@@ -241,6 +244,7 @@ def merge(path1, path2, alpha1, sr, f0, info, name):
"""
opt["sr"] = sr
opt["f0"] = 1 if f0 == "是" else 0
opt["version"] = version
opt["info"] = info
torch.save(opt, "weights/%s.pth" % name)
return "Success."