tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

cuda fix

Browse source
This commit is contained in:
Jannik Streidl 2024-03-22 12:48:48 +01:00
parent fc4e762b05
commit fdef2abdfb
4 changed files with 25 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

3
backend/apps/rag/main.py
View file

@ -71,7 +71,7 @@ from constants import ERROR_MESSAGES
# sentence_transformer_ef = SentenceTransformer(
# model_name_or_path=RAG_EMBEDDING_MODEL,
# cache_folder=RAG_EMBEDDING_MODEL_DIR,
# device=RAG_EMBEDDING_MODEL_DEVICE_TYPE,
# device=DEVICE_TYPE,
# )
@ -178,7 +178,6 @@ async def update_embedding_model(
device=DEVICE_TYPE,
)
)
return {
"status": True,
"embedding_model": app.state.RAG_EMBEDDING_MODEL,