itsclelia commited on
Commit
1365b4a
·
verified ·
1 Parent(s): f3bea3c

fix: make it usable by fastembed

Browse files
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b522e1f0e4cd69533a815eaab4fe7c3005671808179f6d525bfc404347f7e906
3
- size 62509152
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10a13ff3791bf5bbad0c33e5ed3d8d13ed95f0ab92174b5ffdb36026fc7fd965
3
+ size 62509144
scripts/convert-to-safensors.py CHANGED
@@ -12,4 +12,4 @@ d = np.load("model.npz")
12
  W = np.ascontiguousarray(d["rows_fp16"]) # (30522, 1024) float16
13
  assert W.shape == (30522, 1024) and W.dtype == np.float16
14
 
15
- save_file({"embedding.weight": W}, "model.safetensors")
 
12
  W = np.ascontiguousarray(d["rows_fp16"]) # (30522, 1024) float16
13
  assert W.shape == (30522, 1024) and W.dtype == np.float16
14
 
15
+ save_file({"embeddings": W}, "model.safetensors")