Skip to content

Commit 50cfb6b

Browse files
committed
update reranker model
1 parent 842c130 commit 50cfb6b

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

FlagEmbedding/flag_reranker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def __init__(
216216
self,
217217
model_name_or_path: str = None,
218218
use_fp16: bool = False,
219-
use_bf16: bool = True,
219+
use_bf16: bool = False,
220220
cache_dir: str = None,
221221
device: int = 0
222222
) -> None:
@@ -309,7 +309,7 @@ def __init__(
309309
self,
310310
model_name_or_path: str = None,
311311
use_fp16: bool = False,
312-
use_bf16: bool = True,
312+
use_bf16: bool = False,
313313
cache_dir: str = None,
314314
device: int = 0
315315
) -> None:

0 commit comments

Comments
 (0)