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

fix the checking logic in get_model_path

Browse Source 
fix the checking logic in function get_model_path()
This commit is contained in:
Jerry 2024-07-12 14:21:37 +08:00
parent 856845e5f2
commit f07172bfe6
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/apps/rag/utils.py
View File

@ -326,22 +326,25 @@ def get_model_path(model: str, update_model: bool = False):
local_files_only = not update_model local_files_only = not update_model
local_model_path = os.path.join(cache_dir, model)
snapshot_kwargs = { snapshot_kwargs = {
"cache_dir": cache_dir, "cache_dir": cache_dir,
"local_files_only": local_files_only, "local_files_only": local_files_only,
} }
log.debug(f"model: {model}") log.debug(f"model: {model}")
log.debug(f"local_model_path: {local_model_path}")
log.debug(f"snapshot_kwargs: {snapshot_kwargs}") log.debug(f"snapshot_kwargs: {snapshot_kwargs}")
# Inspiration from upstream sentence_transformers # Inspiration from upstream sentence_transformers
if ( if (
os.path.exists(model) os.path.exists(local_model_path)
or ("\\" in model or model.count("/") > 1) or ("\\" in model or model.count("/") > 1)
and local_files_only and local_files_only
): ):
# If fully qualified path exists, return input, else set repo_id # If fully qualified path exists, return input, else set repo_id
return model return local_model_path
elif "/" not in model: elif "/" not in model:
# Set valid repo_id for model short-name # Set valid repo_id for model short-name
model = "sentence-transformers" + "/" + model model = "sentence-transformers" + "/" + model
@ -355,7 +358,7 @@ def get_model_path(model: str, update_model: bool = False):
return model_repo_path return model_repo_path
except Exception as e: except Exception as e:
log.exception(f"Cannot determine model snapshot path: {e}") log.exception(f"Cannot determine model snapshot path: {e}")
return model return local_model_path
def generate_openai_embeddings( def generate_openai_embeddings(