Skip to content

Commit e89654d

Browse files
authored
Hf sources (#429)
* HF sources for all models
1 parent cec8d54 commit e89654d

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

fastembed/text/onnx_embedding.py

+3
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
"license": "mit",
1717
"size_in_GB": 0.42,
1818
"sources": {
19+
"hf": "Qdrant/fast-bge-base-en",
1920
"url": "https://storage.googleapis.com/qdrant-fastembed/fast-bge-base-en.tar.gz",
2021
},
2122
"model_file": "model_optimized.onnx",
@@ -50,6 +51,7 @@
5051
"license": "mit",
5152
"size_in_GB": 0.13,
5253
"sources": {
54+
"hf": "Qdrant/bge-small-en",
5355
"url": "https://storage.googleapis.com/qdrant-fastembed/BAAI-bge-small-en.tar.gz",
5456
},
5557
"model_file": "model_optimized.onnx",
@@ -72,6 +74,7 @@
7274
"license": "mit",
7375
"size_in_GB": 0.09,
7476
"sources": {
77+
"hf": "Qdrant/bge-small-zh-v1.5",
7578
"url": "https://storage.googleapis.com/qdrant-fastembed/fast-bge-small-zh-v1.5.tar.gz",
7679
},
7780
"model_file": "model_optimized.onnx",

fastembed/text/onnx_text_model.py

-1
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,6 @@ def onnx_embed(
7979
onnx_input["token_type_ids"] = np.array(
8080
[np.zeros(len(e), dtype=np.int64) for e in input_ids], dtype=np.int64
8181
)
82-
8382
onnx_input = self._preprocess_onnx_input(onnx_input, **kwargs)
8483

8584
model_output = self.model.run(self.ONNX_OUTPUT_NAMES, onnx_input)

0 commit comments

Comments
 (0)