klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 13:18:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Upgrade default cross-encoder to mixedbread ai's mxbai-rerank-xsmall

Browse Source 
Previous cross-encoder model was a few years old, newer models should
have improved in quality. Model size increases by 50% compared to
previous for better performance, at least on benchmarks
This commit is contained in:
Debanjum Singh Solanky
2024-04-24 09:13:14 +05:30
parent 7eaf9367fe
commit ec41482324
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/helpers.py
View File

@@ -75,7 +75,7 @@ class SearchModelFactory(factory.django.DjangoModelFactory):
name = "default"
model_type = "text"
bi_encoder = "thenlper/gte-small"
cross_encoder = "cross-encoder/ms-marco-MiniLM-L-6-v2"
cross_encoder = "mixedbread-ai/mxbai-rerank-xsmall-v1"
class SubscriptionFactory(factory.django.DjangoModelFactory):