From 2ef71045565b430cdf0e16a7f1a1ebe031e12962 Mon Sep 17 00:00:00 2001 From: goldpulpy Date: Mon, 7 Oct 2024 04:06:42 +0300 Subject: [PATCH] Added more supported models --- README.md | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index f69e652..c726cc5 100644 --- 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.