Skip to content

Commit baebf72

Browse files
committed
update reranker model
1 parent bea26e9 commit baebf72

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

FlagEmbedding/flag_reranker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -324,7 +324,7 @@ def __init__(
324324
self.model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
325325
cache_dir=cache_dir,
326326
trust_remote_code=True,
327-
torch_dtype=torch.bfloat16 if use_bf16 else False)
327+
torch_dtype=torch.bfloat16 if use_bf16 else torch.float32)
328328

329329
self.model_name_or_path = model_name_or_path
330330
self.cache_dir = cache_dir

0 commit comments

Comments
 (0)