chore: prepare repository for public release

2026-08-31 07:23:03 +02:00
parent 03df2db8b6
commit e43d7aca68
164 changed files with 694 additions and 29691 deletions
@@ -13,7 +13,8 @@ def main() -> int:
    parser.add_argument("checkpoint", type=Path)
    args = parser.parse_args()
    import torch
-    checkpoint = torch.load(args.checkpoint, map_location="cpu", weights_only=False)
+
+    checkpoint = torch.load(args.checkpoint, map_location="cpu", weights_only=True)
    payload = {
        key: checkpoint.get(key)
        for key in ("date", "version", "license", "docs", "train_args")