Spaces:
Sleeping
Sleeping
Update retrieverRAG_SF.py
Browse files- retrieverRAG_SF.py +1 -3
retrieverRAG_SF.py
CHANGED
|
@@ -28,9 +28,7 @@ print("Using accelerator device =", device)
|
|
| 28 |
|
| 29 |
|
| 30 |
from sentence_transformers import CrossEncoder
|
| 31 |
-
|
| 32 |
-
model_sf_mxbai = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-12-v1" ,device=device)
|
| 33 |
-
|
| 34 |
|
| 35 |
|
| 36 |
|
|
|
|
| 28 |
|
| 29 |
|
| 30 |
from sentence_transformers import CrossEncoder
|
| 31 |
+
model_sf_mxbai = CrossEncoder("mixedbread-ai/mxbai-rerank-large-v1" ,device=device)
|
|
|
|
|
|
|
| 32 |
|
| 33 |
|
| 34 |
|