Feature Extraction
sentence-transformers
Safetensors
English
code
roberta
python
javascript
go
java
ruby
php
code-search
code-embeddings
multilingual
matryoshka
unixcoder
mteb
Eval Results (legacy)
text-embeddings-inference
Instructions to use divakantpandey/code-embed-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use divakantpandey/code-embed-v2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("divakantpandey/code-embed-v2") 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
Periodic checkpoint at step 1500 (loss=0.00000)
Browse files- backbone/model.safetensors +1 -1
- optimizer.pt +1 -1
- projection_head.pt +1 -1
- rng_state.pt +1 -1
- scheduler.pt +1 -1
- sentence_transformers/2_Dense/model.safetensors +1 -1
- sentence_transformers/3_Dense/model.safetensors +1 -1
- sentence_transformers/README.md +3 -3
- sentence_transformers/model.safetensors +1 -1
- trainer_state.json +2 -2
backbone/model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 503741296
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c20c0675bec09f336d5770f4f3f08e36def580df7cc884eee23a33e96cc0d15
|
| 3 |
size 503741296
|
optimizer.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1031217338
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cebdcdda7c7256442a5c3c776bfaca53d6aa5dfc8f8acccdfad69b41d9a9c5d3
|
| 3 |
size 1031217338
|
projection_head.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 14170320
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ca9f0f4791c27ec4ae99301ad8ded909260d8d66c48d45ca9a65d17708156b26
|
| 3 |
size 14170320
|
rng_state.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 14244
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f07fcc86345d5db87aa51de9f9e5730f0bd97e5c89c3c43e6b35f595c626fc60
|
| 3 |
size 14244
|
scheduler.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1064
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5f73df49ede6a0ccb868a5798d5f5be3a2439265ecbe5b4788805d3df765ae08
|
| 3 |
size 1064
|
sentence_transformers/2_Dense/model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4724896
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e8af18dae773d491719c03414d8540781c232b9bda5ba4e0fcc965301a324f2a
|
| 3 |
size 4724896
|
sentence_transformers/3_Dense/model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 9443488
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:886d631e2a7a3a4628e6d183450cb108b4a7fe3ae9b03ff7682a5c1e6ae142a2
|
| 3 |
size 9443488
|
sentence_transformers/README.md
CHANGED
|
@@ -72,9 +72,9 @@ print(embeddings.shape)
|
|
| 72 |
# Get the similarity scores for the embeddings
|
| 73 |
similarities = model.similarity(embeddings, embeddings)
|
| 74 |
print(similarities)
|
| 75 |
-
# tensor([[1.0000, 0.
|
| 76 |
-
# [0.
|
| 77 |
-
# [0.
|
| 78 |
```
|
| 79 |
<!--
|
| 80 |
### Direct Usage (Transformers)
|
|
|
|
| 72 |
# Get the similarity scores for the embeddings
|
| 73 |
similarities = model.similarity(embeddings, embeddings)
|
| 74 |
print(similarities)
|
| 75 |
+
# tensor([[1.0000, 0.4781, 0.3163],
|
| 76 |
+
# [0.4781, 1.0000, 0.4037],
|
| 77 |
+
# [0.3163, 0.4037, 1.0000]])
|
| 78 |
```
|
| 79 |
<!--
|
| 80 |
### Direct Usage (Transformers)
|
sentence_transformers/model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 503741296
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c20c0675bec09f336d5770f4f3f08e36def580df7cc884eee23a33e96cc0d15
|
| 3 |
size 503741296
|
trainer_state.json
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
{
|
| 2 |
-
"global_step":
|
| 3 |
"current_epoch": 0,
|
| 4 |
"total_training_steps": 6314,
|
| 5 |
-
"learning_rate": 2.
|
| 6 |
"scheduler_name": "cosine",
|
| 7 |
"fp16": false,
|
| 8 |
"bf16": true
|
|
|
|
| 1 |
{
|
| 2 |
+
"global_step": 1500,
|
| 3 |
"current_epoch": 0,
|
| 4 |
"total_training_steps": 6314,
|
| 5 |
+
"learning_rate": 2.8302236490618563e-05,
|
| 6 |
"scheduler_name": "cosine",
|
| 7 |
"fp16": false,
|
| 8 |
"bf16": true
|