Add files via upload

GPT_SoVITS/process_ckpt.py

@@ -0,0 +1,22 @@
+import os
+import sys
+import traceback
+from collections import OrderedDict
+
+import torch
+from 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()