Text Retrieval
Transformers
Safetensors
sentence-transformers
English
kpr-bert
feature-extraction
custom_code
Instructions to use knowledgeable-ai/kpr-retromae with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use knowledgeable-ai/kpr-retromae with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("knowledgeable-ai/kpr-retromae", trust_remote_code=True, device_map="auto") - sentence-transformers
How to use knowledgeable-ai/kpr-retromae with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("knowledgeable-ai/kpr-retromae", trust_remote_code=True) sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -102,8 +102,7 @@ embeddings = model.encode(input_texts)
|
|
| 102 |
print("Embeddings:", embeddings)
|
| 103 |
```
|
| 104 |
|
| 105 |
-
**IMPORTANT:** This code will be supported in versions of Sentence Transformers later than v5.1.0,
|
| 106 |
-
which have not yet been released at the time of writing. Until then, please install the library directly from GitHub:
|
| 107 |
|
| 108 |
```bash
|
| 109 |
pip install git+https://github.com/UKPLab/sentence-transformers.git
|
|
|
|
| 102 |
print("Embeddings:", embeddings)
|
| 103 |
```
|
| 104 |
|
| 105 |
+
**IMPORTANT:** This code will be supported in versions of Sentence Transformers later than v5.1.0, which have not yet been released at the time of writing. Until then, please install the library directly from GitHub:
|
|
|
|
| 106 |
|
| 107 |
```bash
|
| 108 |
pip install git+https://github.com/UKPLab/sentence-transformers.git
|