From 69083b64854ed2e2bbc4702dcf502027a1b4aec5 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Mon, 15 Jul 2024 11:09:05 +0200 Subject: [PATCH] revert --- backend/apps/rag/utils.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/backend/apps/rag/utils.py b/backend/apps/rag/utils.py index 7333592f3..fde89b069 100644 --- a/backend/apps/rag/utils.py +++ b/backend/apps/rag/utils.py @@ -326,25 +326,22 @@ def get_model_path(model: str, update_model: bool = False): local_files_only = not update_model - local_model_path = os.path.join(cache_dir, model) - snapshot_kwargs = { "cache_dir": cache_dir, "local_files_only": local_files_only, } log.debug(f"model: {model}") - log.debug(f"local_model_path: {local_model_path}") log.debug(f"snapshot_kwargs: {snapshot_kwargs}") # Inspiration from upstream sentence_transformers if ( - os.path.exists(local_model_path) + os.path.exists(model) or ("\\" in model or model.count("/") > 1) and local_files_only ): # If fully qualified path exists, return input, else set repo_id - return local_model_path + return model elif "/" not in model: # Set valid repo_id for model short-name model = "sentence-transformers" + "/" + model @@ -358,7 +355,7 @@ def get_model_path(model: str, update_model: bool = False): return model_repo_path except Exception as e: log.exception(f"Cannot determine model snapshot path: {e}") - return local_model_path + return model def generate_openai_embeddings(