https://github.com/davidberenstein1957/fast-sentence-transformers

Simply, faster, sentence-transformers
https://github.com/davidberenstein1957/fast-sentence-transformers

embeddings hacktoberfest nlp onnx sentence-transformers

Last synced: 2 days ago
JSON representation

Simply, faster, sentence-transformers

• Host: GitHub
• URL: https://github.com/davidberenstein1957/fast-sentence-transformers
• Owner: davidberenstein1957
• License: mit
• Created: 2022-04-20T14:03:01.000Z (almost 3 years ago)
• Default Branch: main
• Last Pushed: 2024-08-27T07:38:17.000Z (6 months ago)
• Last Synced: 2024-10-15T04:11:56.385Z (4 months ago)
• Topics: embeddings, hacktoberfest, nlp, onnx, sentence-transformers
• Language: Python
• Homepage:
• Size: 456 KB
• Stars: 135
• Watchers: 5
• Forks: 10
• Open Issues: 3
• Metadata Files:
  • Readme: README.md
  • License: LICENSE
  • Citation: CITATION.cff

Awesome Lists containing this project

README

        
# Fast Sentence Transformers

This repository contains code to run faster feature extractors using tools like quantization, optimization and `ONNX`. Just run your model much faster, while using less of memory. There is not much to it!

[![Python package](https://github.com/Pandora-Intelligence/fast-sentence-transformers/actions/workflows/python-package.yml/badge.svg?branch=main)](https://github.com/Pandora-Intelligence/fast-sentence-transformers/actions/workflows/python-package.yml)

[![Current Release Version](https://img.shields.io/github/release/pandora-intelligence/fast-sentence-transformers.svg?style=flat-square&logo=github)](https://github.com/pandora-intelligence/fast-sentence-transformers/releases)

[![pypi Version](https://img.shields.io/pypi/v/fast-sentence-transformers.svg?style=flat-square&logo=pypi&logoColor=white)](https://pypi.org/project/fast-sentence-transformers/)

[![PyPi downloads](https://static.pepy.tech/personalized-badge/fast-sentence-transformers?period=total&units=international_system&left_color=grey&right_color=orange&left_text=pip%20downloads)](https://pypi.org/project/fast-sentence-transformers/)

[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg?style=flat-square)](https://github.com/ambv/black)

> [Phillip Schmid](https://www.philschmid.de/optimize-sentence-transformers): "We successfully quantized our vanilla Transformers model with Hugging Face and managed to accelerate our model latency from 25.6ms to 12.3ms or 2.09x while keeping 100% of the accuracy on the stsb dataset.

> But I have to say that this isn't a plug and play process you can transfer to any Transformers model, task or dataset.""

## Install

```bash

pip install fast-sentence-transformers

```

Or, for GPU support:

```bash

pip install fast-sentence-transformers[gpu]

```

## Quickstart

```python

from fast_sentence_transformers import FastSentenceTransformer as SentenceTransformer

# use any sentence-transformer

encoder = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", device="cpu")

encoder.encode("Hello hello, hey, hello hello")

encoder.encode(["Life is too short to eat bad food!"] * 2)

```

## Benchmark

Non-exact, indicative benchmark for speed an memory usage with smaller and larger model on `sentence-transformers`

| model                                 | Type   | default | ONNX | ONNX+quantized | ONNX+GPU |

| ------------------------------------- | ------ | ------- | ---- | -------------- | -------- |

| paraphrase-albert-small-v2            | memory | 1x      | 1x   | 1x             | 1x       |

|                                       | speed  | 1x      | 2x   | 5x             | 20x      |

| paraphrase-multilingual-mpnet-base-v2 | memory | 1x      | 1x   | 4x             | 4x       |

|                                       | speed  | 1x      | 2x   | 5x             | 20x      |

## Shout-Out

This package heavily leans on https://www.philschmid.de/optimize-sentence-transformers.