fix: weights folder

2026-01-19 18:41:52 +00:00 · 2023-08-30 00:27:23 +08:00
parent 417c1e58a1
commit 1890584496
5 changed files with 6 additions and 6 deletions
--- a/infer/lib/train/process_ckpt.py
+++ b/infer/lib/train/process_ckpt.py
@@ -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()