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") }