Timothy J. Baek 7 月之前
父節點
當前提交
687cae9b79
共有 1 個文件被更改,包括 1 次插入1 次删除
  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