yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2025-07-20 15:37:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1378 from listeng/add-missing-param

Browse Source 
add missing trust_remote_code[=true] when loading "openbmb/MiniCPM-Reranker-Light"
This commit is contained in:
Jianlyu Chen 2025-04-10 16:29:09 +08:00 committed by GitHub
commit aa0dceee74
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
FlagEmbedding/inference/reranker/encoder_only/base.py
View File

@ -65,6 +65,7 @@ class BaseReranker(AbsReranker):
) )
self.tokenizer = AutoTokenizer.from_pretrained( self.tokenizer = AutoTokenizer.from_pretrained(
model_name_or_path, model_name_or_path,
trust_remote_code=trust_remote_code,
cache_dir=cache_dir cache_dir=cache_dir
) )
self.model = AutoModelForSequenceClassification.from_pretrained( self.model = AutoModelForSequenceClassification.from_pretrained(