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
|
@@ -60,7 +60,7 @@ This model can be used via [Hugging Face Transformers](https://github.com/huggin
|
|
| 60 |
from transformers import AutoTokenizer, AutoModel
|
| 61 |
import torch
|
| 62 |
|
| 63 |
-
MODEL_NAME_OR_PATH = "knowledgeable-ai/kpr-
|
| 64 |
|
| 65 |
input_texts = [
|
| 66 |
"Who founded Dominican Liberation Party?",
|
|
@@ -86,7 +86,7 @@ print("Embeddings:", embeddings)
|
|
| 86 |
```python
|
| 87 |
from sentence_transformers import SentenceTransformer
|
| 88 |
|
| 89 |
-
MODEL_NAME_OR_PATH = "knowledgeable-ai/kpr-
|
| 90 |
|
| 91 |
input_texts = [
|
| 92 |
"Who founded Dominican Liberation Party?",
|
|
|
|
| 60 |
from transformers import AutoTokenizer, AutoModel
|
| 61 |
import torch
|
| 62 |
|
| 63 |
+
MODEL_NAME_OR_PATH = "knowledgeable-ai/kpr-retromae"
|
| 64 |
|
| 65 |
input_texts = [
|
| 66 |
"Who founded Dominican Liberation Party?",
|
|
|
|
| 86 |
```python
|
| 87 |
from sentence_transformers import SentenceTransformer
|
| 88 |
|
| 89 |
+
MODEL_NAME_OR_PATH = "knowledgeable-ai/kpr-retromae"
|
| 90 |
|
| 91 |
input_texts = [
|
| 92 |
"Who founded Dominican Liberation Party?",
|