Update flag_reranker.py

545999961 · web-flow · commit 65cd70d73fc2 · 2024-04-25T14:28:12.000+08:00
can load local layerwise-reranker
diff --git a/FlagEmbedding/flag_reranker.py b/FlagEmbedding/flag_reranker.py
@@ -408,6 +408,7 @@ def __init__(
         self.model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
                                                           cache_dir=cache_dir,
                                                           trust_remote_code=True,
+                                                          local_files_only=True,
                                                           torch_dtype=torch.bfloat16 if use_bf16 else torch.float32)
 
         self.model_name_or_path = model_name_or_path
