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 3500 (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:34a326c9f085b40cf30255b37dfd3a606e565ac12a188e8d52402d0611040f33
|
| 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:3c9321d8e71427fe90931475193ff2ee31ce30375485b1e8bdd6a9eb329e307e
|
| 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:2247c679558817f2a428b22cf8864a340486b2eb8f5fb1b9ee961a2ab899ad27
|
| 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:5d0fd8475b5fbc6eea81d395edb685e11580b77868753597ca132fdf87b58c64
|
| 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:65201ae87736d71f061c694434ad5acc35f3b4a425bcb8df93404c26bdec9108
|
| 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:0ba428cdda5ddb7ea21afbd849bcbfb7d8b70696666f5561ae5633da9ef886d5
|
| 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:1a13770a2a71332fe23165271fa43312060794cbd5a4b831ebd5c84ab7650833
|
| 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.4573, 0.2794],
|
| 76 |
+
# [0.4573, 1.0000, 0.3649],
|
| 77 |
+
# [0.2794, 0.3649, 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:34a326c9f085b40cf30255b37dfd3a606e565ac12a188e8d52402d0611040f33
|
| 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": 3500,
|
| 3 |
"current_epoch": 1,
|
| 4 |
"total_training_steps": 6314,
|
| 5 |
+
"learning_rate": 1.477197658718445e-05,
|
| 6 |
"scheduler_name": "cosine",
|
| 7 |
"fp16": false,
|
| 8 |
"bf16": true
|