Spaces:
Sleeping
Sleeping
File size: 470 Bytes
8223f74 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
parameters:
retriever_type: contextual_reranked # Enable re-ranking
embedding_model_id: text-embedding-3-small
embedding_model_type: openai
embedding_model_dim: 1536
device: mps # or cuda (for Nvidia GPUs) or mps (for Apple M1/M2/M3 chips)
# Re-ranking parameters
enable_reranking: true
rerank_model_name: "cross-encoder/ms-marco-MiniLM-L-2-v2"
stage1_limit: 50 # Retrieve 50 candidates in stage 1
final_k: 10 # Return top 10 after re-ranking
|