Update README.md
Browse files
README.md
CHANGED
@@ -8,6 +8,8 @@ library_name: transformers
|
|
8 |
|
9 |
O4 optimized weights of [`NeuML/pubmedbert-base-embeddings-matryoshka`](https://huggingface.co/NeuML/pubmedbert-base-embeddings-matryoshka).
|
10 |
|
|
|
|
|
11 |
```python
|
12 |
from optimum.onnxruntime import ORTModelForFeatureExtraction
|
13 |
from transformers import AutoTokenizer
|
@@ -39,4 +41,6 @@ embeddings = meanpooling(output, inputs['attention_mask'])
|
|
39 |
dimensions = 256
|
40 |
|
41 |
print("Sentence embeddings:")
|
42 |
-
print(embeddings[:, :dimensions])
|
|
|
|
|
|
8 |
|
9 |
O4 optimized weights of [`NeuML/pubmedbert-base-embeddings-matryoshka`](https://huggingface.co/NeuML/pubmedbert-base-embeddings-matryoshka).
|
10 |
|
11 |
+
## Usage
|
12 |
+
|
13 |
```python
|
14 |
from optimum.onnxruntime import ORTModelForFeatureExtraction
|
15 |
from transformers import AutoTokenizer
|
|
|
41 |
dimensions = 256
|
42 |
|
43 |
print("Sentence embeddings:")
|
44 |
+
print(embeddings[:, :dimensions])
|
45 |
+
|
46 |
+
```
|