diff --git a/README.md b/README.md index 2feb5b721..de5a1ceb3 100644 --- a/README.md +++ b/README.md @@ -86,6 +86,7 @@ Currently, we have released multiple key features, which are listed below to dem - Unified vector storage/indexing of knowledge base - Support for unstructured data such as PDF, TXT, Markdown, CSV, DOC, PPT, and WebURL - Multi LLMs Support, Supports multiple large language models, currently supporting + - 🔥 Baichuan2(7b,13b) - 🔥 Vicuna-v1.5(7b,13b) - 🔥 llama-2(7b,13b,70b) - WizardLM-v1.2(13b) diff --git a/README.zh.md b/README.zh.md index b5d95e432..4df48cf03 100644 --- a/README.zh.md +++ b/README.zh.md @@ -116,6 +116,7 @@ DB-GPT 是一个开源的以数据库为基础的GPT实验项目,使用本地 - 非结构化数据支持包括PDF、MarkDown、CSV、WebURL - 多模型支持 - 支持多种大语言模型, 当前已支持如下模型: + - 🔥 Baichuan2(7b,13b) - 🔥 Vicuna-v1.5(7b,13b) - 🔥 llama-2(7b,13b,70b) - WizardLM-v1.2(13b) diff --git a/pilot/configs/model_config.py b/pilot/configs/model_config.py index 0b2ed520a..2afac5162 100644 --- a/pilot/configs/model_config.py +++ b/pilot/configs/model_config.py @@ -77,6 +77,8 @@ LLM_MODEL_CONFIG = { "baichuan-13b": os.path.join(MODEL_PATH, "Baichuan-13B-Chat"), # please rename "fireballoon/baichuan-vicuna-chinese-7b" to "baichuan-7b" "baichuan-7b": os.path.join(MODEL_PATH, "baichuan-7b"), + "baichuan2-7b": os.path.join(MODEL_PATH, "Baichuan2-7B-Chat"), + "baichuan2-13b": os.path.join(MODEL_PATH, "Baichuan2-13B-Chat"), # (Llama2 based) We only support WizardLM-13B-V1.2 for now, which is trained from Llama-2 13b, see https://huggingface.co/WizardLM/WizardLM-13B-V1.2 "wizardlm-13b": os.path.join(MODEL_PATH, "WizardLM-13B-V1.2"), "llama-cpp": os.path.join(MODEL_PATH, "ggml-model-q4_0.bin"),