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