yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2025-06-27 02:39:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

update reranker inference

Browse Source
This commit is contained in:
cfli 2025-05-28 16:14:32 +08:00
parent bf410d51e9
commit 018b984e23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
research/Matroyshka_reranker/inference/rank_model.py
View File

@ -96,7 +96,8 @@ class MatroyshkaReranker(AbsReranker):
self.tokenizer = AutoTokenizer.from_pretrained( self.tokenizer = AutoTokenizer.from_pretrained(
model_name_or_path, model_name_or_path,
cache_dir=cache_dir, cache_dir=cache_dir,
trust_remote_code=trust_remote_code trust_remote_code=trust_remote_code,
use_fast=False,
) )
self.tokenizer.padding_side = 'right' self.tokenizer.padding_side = 'right'