mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-13 13:10:29 +00:00
chore:v0.5.8 tag (#1632)
This commit is contained in:
@@ -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)
|
||||
|
@@ -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."""
|
||||
|
Reference in New Issue
Block a user