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 750 (loss=0.00006)
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:2d4fc472222aead801b3982bffea882dc54629d4ee9d730bd6b584bde5ae2f62
|
| 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:831c9d6b7b6ea5cd831ccf60d0213d8031eab866e8e620d044ee3deef21acd36
|
| 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:e2d08233331ba64e97f4e420267484450a468f95cf06cb9e936b0ded1e4910f5
|
| 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:68572cb2c95ac180f2e71e3210428faa4f91d673f18ec86f667645267e2a517b
|
| 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:31bbcd7f1a25decd8fb6cdce5a47aef530f3659f82ee1a26d8640be8446be6dc
|
| 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:5227655877490863b5f80c46d749bcbeea1a96cb388a9037130ba99f8b06b878
|
| 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:5d4d375cc8aded73d3ce2eb241e25383f8f84dc0c80ddb5822e182d4699a6ef1
|
| 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.5272, 0.4017],
|
| 76 |
+
# [0.5272, 1.0000, 0.4583],
|
| 77 |
+
# [0.4017, 0.4583, 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:2d4fc472222aead801b3982bffea882dc54629d4ee9d730bd6b584bde5ae2f62
|
| 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": 750,
|
| 3 |
"current_epoch": 0,
|
| 4 |
"total_training_steps": 6314,
|
| 5 |
+
"learning_rate": 2.9967555357291773e-05,
|
| 6 |
"scheduler_name": "cosine",
|
| 7 |
"fp16": false,
|
| 8 |
"bf16": true
|