From 54cf79f521494cc67e4b2fe6f5a2ca45b79914f1 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Wed, 6 Sep 2023 21:17:32 +0800 Subject: [PATCH] add support for new embedding models sensenova/piccolo-base-zh and sensenova/piccolo-large-zh --- README.md | 2 ++ configs/model_config.py.example | 2 ++ 2 files changed, 4 insertions(+) diff --git a/README.md b/README.md index 646dfc7..1090ac9 100644 --- a/README.md +++ b/README.md @@ -127,6 +127,8 @@ docker run -d --gpus all -p 80:8501 registry.cn-beijing.aliyuncs.com/chatchat/ch - [BAAI/bge-base-zh](https://huggingface.co/BAAI/bge-base-zh) - [BAAI/bge-large-zh](https://huggingface.co/BAAI/bge-large-zh) - [BAAI/bge-large-zh-noinstruct](https://huggingface.co/BAAI/bge-large-zh-noinstruct) +- [sensenova/piccolo-base-zh](https://huggingface.co/sensenova/piccolo-base-zh) +- [sensenova/piccolo-large-zh](https://huggingface.co/sensenova/piccolo-large-zh) - [shibing624/text2vec-base-chinese-sentence](https://huggingface.co/shibing624/text2vec-base-chinese-sentence) - [shibing624/text2vec-base-chinese-paraphrase](https://huggingface.co/shibing624/text2vec-base-chinese-paraphrase) - [shibing624/text2vec-base-multilingual](https://huggingface.co/shibing624/text2vec-base-multilingual) diff --git a/configs/model_config.py.example b/configs/model_config.py.example index de5da00..39c51ca 100644 --- a/configs/model_config.py.example +++ b/configs/model_config.py.example @@ -26,6 +26,8 @@ embedding_model_dict = { "bge-base-zh": "BAAI/bge-base-zh", "bge-large-zh": "BAAI/bge-large-zh", "bge-large-zh-noinstruct": "BAAI/bge-large-zh-noinstruct", + "piccolo-base-zh": "sensenova/piccolo-base-zh", + "piccolo-large-zh": "sensenova/piccolo-large-zh", "text-embedding-ada-002": os.environ.get("OPENAI_API_KEY") }