Feature Extraction
sentence-transformers
Safetensors
Transformers
English
qwen3
sentence-similarity
retrieval
agent-skills
skill-routing
skillcorpus
contrastive-learning
text-embeddings-inference
Instructions to use EverMind-AI/skillrep-embedding-0.6b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use EverMind-AI/skillrep-embedding-0.6b with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("EverMind-AI/skillrep-embedding-0.6b") 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] - Transformers
How to use EverMind-AI/skillrep-embedding-0.6b with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="EverMind-AI/skillrep-embedding-0.6b")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("EverMind-AI/skillrep-embedding-0.6b") model = AutoModel.from_pretrained("EverMind-AI/skillrep-embedding-0.6b", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Query prompt: one space after the instruction, as used in the paper
Browse files- README.md +4 -4
- config_sentence_transformers.json +1 -1
README.md
CHANGED
|
@@ -43,9 +43,9 @@ before that) or `sentence-transformers>=3.0`.
|
|
| 43 |
## Usage
|
| 44 |
|
| 45 |
The two sides are encoded asymmetrically: a task description carries an
|
| 46 |
-
instruction prefix
|
| 47 |
-
|
| 48 |
-
|
| 49 |
|
| 50 |
```python
|
| 51 |
import torch
|
|
@@ -58,7 +58,7 @@ model = AutoModel.from_pretrained(MODEL, dtype=torch.bfloat16).cuda().eval()
|
|
| 58 |
|
| 59 |
QUERY_INSTRUCTION = (
|
| 60 |
"Instruct: Given a coding task description, retrieve the most relevant "
|
| 61 |
-
"skill document that would help an agent complete the task\nQuery:"
|
| 62 |
)
|
| 63 |
|
| 64 |
|
|
|
|
| 43 |
## Usage
|
| 44 |
|
| 45 |
The two sides are encoded asymmetrically: a task description carries an
|
| 46 |
+
instruction prefix followed by one space, and a skill is its bare body. Encode
|
| 47 |
+
them this way or retrieval quality drops. Embeddings come back L2-normalized, so
|
| 48 |
+
cosine similarity is a plain dot product.
|
| 49 |
|
| 50 |
```python
|
| 51 |
import torch
|
|
|
|
| 58 |
|
| 59 |
QUERY_INSTRUCTION = (
|
| 60 |
"Instruct: Given a coding task description, retrieve the most relevant "
|
| 61 |
+
"skill document that would help an agent complete the task\nQuery: "
|
| 62 |
)
|
| 63 |
|
| 64 |
|
config_sentence_transformers.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"prompts": {
|
| 3 |
-
"query": "Instruct: Given a coding task description, retrieve the most relevant skill document that would help an agent complete the task\nQuery:",
|
| 4 |
"document": ""
|
| 5 |
},
|
| 6 |
"default_prompt_name": null,
|
|
|
|
| 1 |
{
|
| 2 |
"prompts": {
|
| 3 |
+
"query": "Instruct: Given a coding task description, retrieve the most relevant skill document that would help an agent complete the task\nQuery: ",
|
| 4 |
"document": ""
|
| 5 |
},
|
| 6 |
"default_prompt_name": null,
|