Fix up docs (#477)

This commit is contained in:
Chang She
2023-09-05 22:29:50 -07:00
committed by GitHub
parent 9a9a73a65d
commit 32163063dc
5 changed files with 18 additions and 2 deletions

View File

@@ -17,5 +17,6 @@ from .functions import (
EmbeddingFunctionModel,
EmbeddingFunctionRegistry,
SentenceTransformerEmbeddingFunction,
TextEmbeddingFunctionModel,
)
from .utils import with_embeddings

View File

@@ -169,6 +169,10 @@ class TextEmbeddingFunctionModel(EmbeddingFunctionModel):
@REGISTRY.register()
class SentenceTransformerEmbeddingFunction(TextEmbeddingFunctionModel):
"""
An embedding function that uses the sentence-transformers library
"""
name: str = "all-MiniLM-L6-v2"
device: str = "cpu"
normalize: bool = False