fix: weights folder

This commit is contained in:
源文雨 2023-08-30 00:27:23 +08:00
parent 417c1e58a1
commit 1890584496
5 changed files with 6 additions and 6 deletions

View File

@ -1,5 +1,5 @@
{
"pth_path": "weights/kikiV1.pth",
"pth_path": "assets/weights/kikiV1.pth",
"index_path": "logs/kikiV1.index",
"sg_input_device": "VoiceMeeter Output (VB-Audio Vo (MME)",
"sg_output_device": "VoiceMeeter Aux Input (VB-Audio (MME)",

View File

@ -36,7 +36,7 @@ shutil.rmtree("%s/runtime/Lib/site-packages/infer_pack" % (now_dir), ignore_erro
shutil.rmtree("%s/runtime/Lib/site-packages/uvr5_pack" % (now_dir), ignore_errors=True)
os.makedirs(tmp, exist_ok=True)
os.makedirs(os.path.join(now_dir, "logs"), exist_ok=True)
os.makedirs(os.path.join(now_dir, "weights"), exist_ok=True)
os.makedirs(os.path.join(now_dir, "assets/weights"), exist_ok=True)
warnings.filterwarnings("ignore")
torch.manual_seed(114514)

View File

@ -197,7 +197,7 @@ def change_info(path, info, name):
ckpt["info"] = info
if name == "":
name = os.path.basename(path)
torch.save(ckpt, "weights/%s" % name)
torch.save(ckpt, "assets/weights/%s" % name)
return "Success."
except:
return traceback.format_exc()

View File

@ -226,7 +226,7 @@ class Predictor:
class MDXNetDereverb:
def __init__(self, chunks, device):
self.onnx = "uvr5_weights/onnx_dereverb_By_FoxJoy"
self.onnx = "assets/uvr5_weights/onnx_dereverb_By_FoxJoy"
self.shifts = 10 # 'Predict with randomised equivariant stabilisation'
self.mixing = "min_mag" # ['default','min_mag','max_mag']
self.chunks = chunks

View File

@ -89,6 +89,6 @@ def main(path, root):
if __name__ == "__main__":
query_path = r"weights\mi v3.pth"
reference_root = r"weights"
query_path = r"assets\weights\mi v3.pth"
reference_root = r"assets\weights"
main(query_path, reference_root)