chore: fix get_registry call in baai embeddings example (#1230)

This commit is contained in:
Alex Kohler
2024-04-19 21:55:16 -04:00
committed by GitHub
parent 1e5ccb1614
commit c1a7d65473

View File

@@ -159,7 +159,7 @@ Allows you to set parameters when registering a `sentence-transformers` object.
from lancedb.embeddings import get_registry
db = lancedb.connect("/tmp/db")
model = get_registry.get("sentence-transformers").create(name="BAAI/bge-small-en-v1.5", device="cpu")
model = get_registry().get("sentence-transformers").create(name="BAAI/bge-small-en-v1.5", device="cpu")
class Words(LanceModel):
text: str = model.SourceField()