Timothy J. Baek 7 months ago
parent
commit
687cae9b79
1 changed files with 1 additions and 1 deletions
  1. 1 1
      backend/open_webui/apps/rag/main.py

+ 1 - 1
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