Added more supported models

goldpulpy · Oct 7, 2024 · 2ef7104 · 2ef7104
1 parent 7713b0c
commit 2ef7104
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -36,10 +36,18 @@ You don't need to download anything; the package itself will download the model
 
 Below are the models currently added to the special repository, including their file size and a link to the source.
 
-| Model             | Parameters | FP32   | FP16  | INT8  | Source link                                                             |
-| ----------------- | ---------- | ------ | ----- | ----- | ----------------------------------------------------------------------- |
-| all-MiniLM-L6-v2  | 22.7M      | 90 MB  | 45 MB | 23 MB | [HF](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) 🤗  |
-| all-MiniLM-L12-v2 | 33.4M      | 127 MB | 65 MB | 32 MB | [HF](https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2) 🤗 |
+| Model                                 | Parameters | FP32   | FP16  | INT8  | Source link                                                                                 |
+| ------------------------------------- | ---------- | ------ | ----- | ----- | ------------------------------------------------------------------------------------------- |
+| all-MiniLM-L6-v2                      | 22.7M      | 90MB   | 45MB  | 23MB  | [HF](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) 🤗                      |
+| paraphrase-MiniLM-L6-v2               | 22.7M      | 90MB   | 45MB  | 23MB  | [HF](https://huggingface.co/sentence-transformers/paraphrase-MiniLM-L6-v2) 🤗               |
+| all-MiniLM-L12-v2                     | 33.4M      | 127MB  | 65MB  | 32MB  | [HF](https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2) 🤗                     |
+| gte-small                             | 33.4M      | 127MB  | 65MB  | 32MB  | [HF](https://huggingface.co/thenlper/gte-small) 🤗                                          |
+| all-mpnet-base-v2                     | 109M       | 418MB  | 209MB | 105MB | [HF](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) 🤗                     |
+| paraphrase-multilingual-MiniLM-L12-v2 | 118M       | 449MB  | 225MB | 113MB | [HF](https://huggingface.co/sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2) 🤗 |
+| text2vec-base-multilingual            | 118M       | 449MB  | 225MB | 113MB | [HF](https://huggingface.co/shibing624/text2vec-base-multilingual) 🤗                       |
+| gte-multilingual-base                 | 305M       | 1.17GB | 599MB | 324MB | [HF](https://huggingface.co/Alibaba-NLP/gte-multilingual-base) 🤗                           |
+| gte-large                             | 335M       | 1.25GB | 640MB | 321MB | [HF](https://huggingface.co/thenlper/gte-large) 🤗                                          |
+| LaBSE                                 | 470M       | 1.75GB | 898MB | 450MB | [HF](https://huggingface.co/sentence-transformers/LaBSE) 🤗                                 |
 
 **pysentence-similarity** supports `FP32`, `FP16`, and `INT8` dtypes.