switch to 384 vector model BAAI/bge-small-en-v1.5

toranb · Nov 11, 2023 · f1a3f37 · f1a3f37
1 parent af53f8d
commit f1a3f37
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/demo/application.ex b/lib/demo/application.ex
@@ -36,7 +36,7 @@ defmodule Demo.Application do
   end
 
   def serving() do
-    repo = "thenlper/gte-base"
+    repo = "BAAI/bge-small-en-v1.5"
     {:ok, model_info} = Bumblebee.load_model({:hf, repo})
     {:ok, tokenizer} = Bumblebee.load_tokenizer({:hf, repo})
 

diff --git a/priv/repo/migrations/20231104123422_add_document_section.exs b/priv/repo/migrations/20231104123422_add_document_section.exs
@@ -12,7 +12,7 @@ defmodule Demo.Repo.Migrations.AddDocumentSection do
     create table(:sections) do
       add :page, :integer, null: false
       add :text, :text, null: false
-      add :embedding, :vector, size: 768
+      add :embedding, :vector, size: 384
 
       add :document_id, references(:documents), null: false