From b3f83060c8530400418054059527f333c89e7dea Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Tue, 25 Apr 2023 20:14:33 +0800 Subject: [PATCH] update model_config.py --- configs/model_config.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/configs/model_config.py b/configs/model_config.py index 9d66d374..99c4ddc0 100644 --- a/configs/model_config.py +++ b/configs/model_config.py @@ -1,6 +1,6 @@ import torch.cuda import torch.backends - +import os embedding_model_dict = { "ernie-tiny": "nghuyong/ernie-3.0-nano-zh", @@ -31,7 +31,7 @@ USE_PTUNING_V2 = False # LLM running device LLM_DEVICE = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu" -VS_ROOT_PATH = "./vector_store/" +VS_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "vector_store") -UPLOAD_ROOT_PATH = "./content/" +UPLOAD_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "content")