Update README.md
Browse files
README.md
CHANGED
@@ -35,7 +35,12 @@ from transformers import AutoTokenizer
|
|
35 |
tokenizer = AutoTokenizer.from_pretrained("ddmitov/bge_m3_dense_colbert_onnx")
|
36 |
ort_session = ort.InferenceSession("model.onnx")
|
37 |
|
38 |
-
inputs = tokenizer(
|
|
|
|
|
|
|
|
|
|
|
39 |
inputs_onnx = {k: ort.OrtValue.ortvalue_from_numpy(v) for k, v in inputs.items()}
|
40 |
|
41 |
outputs = ort_session.run(None, inputs_onnx)
|
|
|
35 |
tokenizer = AutoTokenizer.from_pretrained("ddmitov/bge_m3_dense_colbert_onnx")
|
36 |
ort_session = ort.InferenceSession("model.onnx")
|
37 |
|
38 |
+
inputs = tokenizer(
|
39 |
+
"BGE M3 is an embedding model supporting dense retrieval, lexical matching and multi-vector interaction.",
|
40 |
+
padding="longest",
|
41 |
+
return_tensors="np"
|
42 |
+
)
|
43 |
+
|
44 |
inputs_onnx = {k: ort.OrtValue.ortvalue_from_numpy(v) for k, v in inputs.items()}
|
45 |
|
46 |
outputs = ort_session.run(None, inputs_onnx)
|