UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-06 13:10:16 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-19 17:31:59 +02:00
parent f8fffdd288
commit 687cae9b79
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/open_webui/apps/rag/main.py
View File

@ -206,7 +206,7 @@ def update_reranking_model(
self.device = "cuda" if torch.cuda.is_available() else "cpu" self.device = "cuda" if torch.cuda.is_available() else "cpu"
self.ckpt = Checkpoint( self.ckpt = Checkpoint(
get_model_path(name, auto_update), get_model_path(name, auto_update),
colbert_config=ColBERTConfig(), colbert_config=ColBERTConfig(model_name=name),
).to(self.device) ).to(self.device)
pass pass