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