chore:v0.5.8 tag (#1632)

This commit is contained in:
Aries-ckt
2024-06-13 17:53:46 +08:00
committed by GitHub
parent 0541d1494c
commit 57f7b6d64c
5 changed files with 4 additions and 4 deletions

View File

@@ -294,7 +294,7 @@ class HuggingFaceBgeEmbeddings(BaseModel, Embeddings):
kwargs["client"] = sentence_transformers.SentenceTransformer(
kwargs.get("model_name"),
cache_folder=kwargs.get("cache_folder"),
**kwargs.get("model_kwargs"),
**(kwargs.get("model_kwargs") or {}),
)
super().__init__(**kwargs)

View File

@@ -17,7 +17,7 @@ class CrossEncoderRerankEmbeddings(BaseModel, RerankEmbeddings):
client: Any #: :meta private:
model_name: str = "BAAI/bge-reranker-base"
max_length: int
max_length: int = None # type: ignore
"""Max length for input sequences. Longer sequences will be truncated. If None, max
length of the model will be used"""
"""Model name to use."""