refac

2025-01-29 13:58:09 +00:00 · 2024-09-19 17:31:59 +02:00 · 2024-09-19 17:31:59 +02:00 · 687cae9b79
commit 687cae9b79
parent f8fffdd288
1 changed files with 1 additions and 1 deletions
--- 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