From 687cae9b793eb7ee0c0b4dbf9e12ecccbc01b00a Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Thu, 19 Sep 2024 17:31:59 +0200 Subject: [PATCH] refac --- backend/open_webui/apps/rag/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/open_webui/apps/rag/main.py b/backend/open_webui/apps/rag/main.py index e10c03bc3..10ed50f53 100644 --- a/backend/open_webui/apps/rag/main.py +++ b/backend/open_webui/apps/rag/main.py @@ -206,7 +206,7 @@ def update_reranking_model( self.device = "cuda" if torch.cuda.is_available() else "cpu" self.ckpt = Checkpoint( get_model_path(name, auto_update), - colbert_config=ColBERTConfig(), + colbert_config=ColBERTConfig(model_name=name), ).to(self.device) pass