CodeMind / tests /test_gemma.py
devjas1
(TESTS): add initial test for embeddinggemma-300m model encoding
fadc4bc
raw
history blame
194 Bytes
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("./models/embeddinggemma-300m")
embedding = model.encode("This is a test string.")
print(embedding.shape)