local_copilot / data /testa.py
Kash6's picture
Deploy AI Coding Assistant
04653e2
raw
history blame contribute delete
244 Bytes
from sentence_transformers import SentenceTransformer
model = SentenceTransformer('all-MiniLM-L6-v2', device='cpu')
print("Model loaded")
texts = ["def foo(): return 1", "class A: pass"]
embeddings = model.encode(texts)
print("Embedding done")