Update README.md
Browse files
README.md
CHANGED
|
@@ -1,6 +1,11 @@
|
|
| 1 |
---
|
| 2 |
license: unlicense
|
| 3 |
pipeline_tag: sentence-similarity
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
---
|
| 5 |
Encoder-model for search query similarity task.
|
| 6 |
|
|
@@ -27,4 +32,4 @@ model.eval()
|
|
| 27 |
vector = model(input_ids=input_ids, attention_mask=input_ids>tokenizer.mask_token_id)[0][0,0]
|
| 28 |
|
| 29 |
assert model.config.hidden_size == vector.shape[0]
|
| 30 |
-
```
|
|
|
|
| 1 |
---
|
| 2 |
license: unlicense
|
| 3 |
pipeline_tag: sentence-similarity
|
| 4 |
+
language:
|
| 5 |
+
- ru
|
| 6 |
+
tags:
|
| 7 |
+
- PyTorch
|
| 8 |
+
- Transformers
|
| 9 |
---
|
| 10 |
Encoder-model for search query similarity task.
|
| 11 |
|
|
|
|
| 32 |
vector = model(input_ids=input_ids, attention_mask=input_ids>tokenizer.mask_token_id)[0][0,0]
|
| 33 |
|
| 34 |
assert model.config.hidden_size == vector.shape[0]
|
| 35 |
+
```
|