From def9fd833059eb84720b99b38a4f9566c7feb05a Mon Sep 17 00:00:00 2001
From: aries-ckt <916701291@qq.com>
Date: Wed, 17 May 2023 22:24:25 +0800
Subject: [PATCH] fix:update model name

---
 pilot/configs/model_config.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pilot/configs/model_config.py b/pilot/configs/model_config.py
index 819152845..128e87513 100644
--- a/pilot/configs/model_config.py
+++ b/pilot/configs/model_config.py
@@ -20,7 +20,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 LLM_MODEL_CONFIG = {
     "flan-t5-base": os.path.join(MODEL_PATH, "flan-t5-base"),
     "vicuna-13b": os.path.join(MODEL_PATH, "vicuna-13b"),
-    "text2vec": os.path.join(MODEL_PATH, "text2vec"),
+    "text2vec": os.path.join(MODEL_PATH, "text2vec-large-chinese"),
     "sentence-transforms": os.path.join(MODEL_PATH, "all-MiniLM-L6-v2")
 }