devjas1
commited on
Commit
·
aeb8d29
1
Parent(s):
cdd329f
(TESTS): add initial test for embeddinggemma-300m model encoding
Browse files- .gitattributes +2 -0
- tests/test_gemma.py +5 -0
.gitattributes
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.gguf filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
C:/Users/xJB6x/Projects/CodeMind/models/embeddinggemma-300m/* filter=lfs diff=lfs merge=lfs -text
|
tests/test_gemma.py
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from sentence_transformers import SentenceTransformer
|
| 2 |
+
|
| 3 |
+
model = SentenceTransformer("./models/embeddinggemma-300m")
|
| 4 |
+
embedding = model.encode("This is a test string.")
|
| 5 |
+
print(embedding.shape)
|