fix: make it usable by fastembed
Browse files- model.safetensors +2 -2
- scripts/convert-to-safensors.py +1 -1
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 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({"
|
|
|
|
| 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")
|