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 4000 (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:7683275d6ca16f9edf6c136da746d6344798a1306b6804ab66cc71091292a60b
|
| 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:1f73ba2aebb0d3f7cff180fb12585c3fe66157f21e9378ae556f97a842b64b15
|
| 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:b1597c9d0a6caa27dbd3c410f80733e866217e031518ac595e46410f13b92a2e
|
| 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:5c224d32193d70443f02c7712d85907bb1bc89d6f41b37f0c11ff64b55824ea3
|
| 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:2d07a32c0fe960a7e278d3ec6692d623315ed7907d2e575d7d34f43f51aeec06
|
| 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:b7990c5a014330348df425fee8ed50fb1efe42483f9d6b4f2ee92014e14a7d63
|
| 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:4002306dbf61db3ad27807dcb12bbbf8864f41ff6f1d0f051ae2256556326384
|
| 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.4405, 0.2547],
|
| 76 |
+
# [0.4405, 1.0000, 0.3439],
|
| 77 |
+
# [0.2547, 0.3439, 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:7683275d6ca16f9edf6c136da746d6344798a1306b6804ab66cc71091292a60b
|
| 3 |
size 503741296
|
trainer_state.json
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
{
|
| 2 |
-
"global_step":
|
| 3 |
"current_epoch": 1,
|
| 4 |
"total_training_steps": 6314,
|
| 5 |
-
"learning_rate": 1.
|
| 6 |
"scheduler_name": "cosine",
|
| 7 |
"fp16": false,
|
| 8 |
"bf16": true
|
|
|
|
| 1 |
{
|
| 2 |
+
"global_step": 4000,
|
| 3 |
"current_epoch": 1,
|
| 4 |
"total_training_steps": 6314,
|
| 5 |
+
"learning_rate": 1.0687655192808827e-05,
|
| 6 |
"scheduler_name": "cosine",
|
| 7 |
"fp16": false,
|
| 8 |
"bf16": true
|