formatting

pull/10242/head
olgavrou 1 year ago
parent 67dc1a9dd2
commit 4e9aecda90

@ -60,6 +60,7 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]):
if model is None:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("all-mpnet-base-v2")
self.model = model

Loading…
Cancel
Save