new checkpoint, safetensor torch

589-20240113-071533.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8d0bcec2b0181b164a34572c12fecf766321cfdd63b96b551e9345532ebae3da
+size 663426

patch_weights.py

@@ -1,18 +1,16 @@
 import numpy as np
-import mlx.core as mx
+import torch
 from glob import glob
-from safetensors.numpy import save_file
+from safetensors.torch import save_file, load_file
 
-patch_weights = mx.load("49-20240112-184735.npz")
+patch_weights = np.load("589-20240113-071533.npz")
 
 for file in glob("model*.safetensors"):
     print(f"{file=}")
-    weights = mx.load(file)
-    for k, v in weights.items():
+    weights = load_file(file)
+    for k, tensor in weights.items():
         if k in patch_weights:
             print(f"patching {k}")
-            weights[k] = np.array(patch_weights[k], copy=False)
-        else:
-            weights[k] = np.array(v, copy=False)
+            weights[k] = torch.from_numpy(patch_weights[k])
     save_file(weights, "patched_" + file)