Add files via upload

This commit is contained in:
YangQianli92 2024-04-18 11:44:29 +08:00 committed by GitHub
parent d5d45114a4
commit 4777fead2c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -124,7 +124,7 @@ class CohereRerankConfig(BaseRankerConfig):
api_key: str = Field(default="YOUR_COHERE_API")
class FlagEmbeddingConfig(BaseRankerConfig):
class BGERerankConfig(BaseRankerConfig):
model: str = Field(default="BAAI/bge-reranker-large", description="BAAI Reranker model name.")
use_fp16: bool = Field(default=True, description="Whether to use fp16 for inference.")