fix: weights folder
This commit is contained in:
parent
417c1e58a1
commit
1890584496
@ -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)",
|
||||
|
@ -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)
|
||||
|
||||
|
@ -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()
|
||||
|
@ -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
|
||||
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user