add reranker model to baseline infinity service

jpetrucciani · Jan 2, 2025 · 0e23017 · 0e23017
1 parent 5e2de66
commit 0e23017
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/hosts/modules/servers/infinity.nix b/hosts/modules/servers/infinity.nix
@@ -30,6 +30,7 @@ in
       type = listOf str;
       default = [
         "nomic-ai/nomic-embed-text-v1.5"
+        "mixedbread-ai/mxbai-rerank-base-v1"
         # check out the MTEB leaderboard!
         ### https://huggingface.co/spaces/mteb/leaderboard
         # "jinaai/jina-embeddings-v3"

diff --git a/hosts/terra/api.nix b/hosts/terra/api.nix
@@ -45,7 +45,7 @@ let
     oauth2client
     peewee
     psycopg2
-    pydrive2
+    # pydrive2
 
     # server
     fastapi