yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2026-01-06 12:11:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

update desc

Browse Source
This commit is contained in:
cfli 2024-10-22 21:20:08 +08:00
parent 07eef8c828
commit c038a8f962
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
FlagEmbedding/abc/inference/AbsReranker.py
View File

@ -17,7 +17,7 @@ logger = logging.getLogger(__name__)
class AbsReranker(ABC):
"""
Base class for embedder.
Extend this class and implement `encode_queries`, `encode_passages`, `encode` for custom embedders.
Extend this class and implement `compute_score_single_gpu` for custom rerankers.
"""
def __init__(

2
FlagEmbedding/inference/reranker/model_mapping.py
View File

@ -38,5 +38,5 @@ MODEL_MAPPING = OrderedDict([
"bge-reranker-v2.5-gemma2-lightweight",
RerankerConfig(LightWeightFlagLLMReranker)
),
# TODO: Add more models, such as Jina, Stella_v5, NV-Embed, etc.
# TODO: Add more models, such as Jina, e5, etc.
])