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

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-19 18:40:23 +02:00
parent a2b77fd072
commit cba2d31175
3 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -267,7 +267,9 @@ def update_reranking_model(
return scores return scores
try: try:
app.state.sentence_transformer_rf = ColBERT(reranking_model) app.state.sentence_transformer_rf = ColBERT(
get_model_path(reranking_model, auto_update)
)
except Exception as e: except Exception as e:
log.error(f"ColBERT: {e}") log.error(f"ColBERT: {e}")
app.state.sentence_transformer_rf = None app.state.sentence_transformer_rf = None

2
backend/requirements.txt
View File

@ -44,6 +44,8 @@ pymilvus==2.4.6
sentence-transformers==3.0.1 sentence-transformers==3.0.1
colbert-ai==0.2.21 colbert-ai==0.2.21
einop==0.8.0
flash_attn==2.6.3
pypdf==4.3.1 pypdf==4.3.1
docx2txt==0.8 docx2txt==0.8

2
pyproject.toml
View File

@ -51,6 +51,8 @@ dependencies = [
"sentence-transformers==3.0.1", "sentence-transformers==3.0.1",
"colbert-ai==0.2.21", "colbert-ai==0.2.21",
"einop==0.8.0",
"flash_attn==2.6.3",
"pypdf==4.3.1", "pypdf==4.3.1",
"docx2txt==0.8", "docx2txt==0.8",