File size: 354 Bytes
1026698 | 1 2 3 4 5 6 7 8 9 10 | from sentence_transformers import SentenceTransformer
import time, numpy as np
model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
texts = ["This is a test."] * 1000 # batch you plan to use
t0 = time.time()
emb = model.encode(texts, convert_to_numpy=True)
t1 = time.time()
print("seconds for 1000:", t1-t0)
print("shape:", emb.shape)
|