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

add missing trust_remote_code[=true] when loading "openbmb/MiniCPM-Reranker-Light"
2025-06-27 02:39:58 +00:00 · 2025-04-10 16:29:09 +08:00 · 2025-04-10 16:29:09 +08:00 · aa0dceee74
commit aa0dceee74
parent c32592e53e 9a2334bf5f
1 changed files with 1 additions and 0 deletions
--- a/FlagEmbedding/inference/reranker/encoder_only/base.py
+++ b/FlagEmbedding/inference/reranker/encoder_only/base.py
@ -65,6 +65,7 @@ class BaseReranker(AbsReranker):
        )
        self.tokenizer = AutoTokenizer.from_pretrained(
            model_name_or_path, 
+            trust_remote_code=trust_remote_code, 
            cache_dir=cache_dir
        )
        self.model = AutoModelForSequenceClassification.from_pretrained(