diff --git a/README.md b/README.md index e210947..f29b499 100644 --- a/README.md +++ b/README.md @@ -93,6 +93,10 @@ - [moka-ai/m3e-small](https://huggingface.co/moka-ai/m3e-small) - [moka-ai/m3e-base](https://huggingface.co/moka-ai/m3e-base) +- [moka-ai/m3e-large](https://huggingface.co/moka-ai/m3e-large) +- [BAAI/bge-small-zh](https://huggingface.co/BAAI/bge-small-zh) +- [BAAI/bge-base-zh](https://huggingface.co/BAAI/bge-base-zh) +- [BAAI/bge-large-zh](https://huggingface.co/BAAI/bge-large-zh) - [text2vec-base-chinese-sentence](https://huggingface.co/shibing624/text2vec-base-chinese-sentence) - [text2vec-base-chinese-paraphrase](https://huggingface.co/shibing624/text2vec-base-chinese-paraphrase) - [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 75ecacc..e26fa50 100644 --- a/configs/model_config.py.example +++ b/configs/model_config.py.example @@ -25,6 +25,9 @@ embedding_model_dict = { "m3e-small": "moka-ai/m3e-small", "m3e-base": "moka-ai/m3e-base", "m3e-large": "moka-ai/m3e-large", + "bge-small-zh": "BAAI/bge-small-zh", + "bge-base-zh": "BAAI/bge-base-zh", + "bge-large-zh": "BAAI/bge-large-zh" } # 选用的 Embedding 名称