Skip to main content

Deploy SentenceTransformers embedding models to a ray cluster

Project description

ray-embedding

A Python library for deploying SentenceTransformers models to a ray cluster. This tool encapsulates inference logic that uses SentenceTransformers to load any compatible embedding model from the Hugging Face hub and compute embeddings for input text.

This library is meant to be used with the embedding-models Ray cluster.

Refer to this Ray Serve deployment config to see how this library is used.

Supports the following backends

  • pytorch-gpu
  • pytorch-cpu

Planned:

  • onnx-gpu
  • onnx-cpu
  • openvino-cpu
  • fastembed-onnx-cpu

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ray_embedding-0.11.1.tar.gz (4.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

ray_embedding-0.11.1-py3-none-any.whl (5.9 kB view details)

Uploaded Python 3

File details

Details for the file ray_embedding-0.11.1.tar.gz.

File metadata

  • Download URL: ray_embedding-0.11.1.tar.gz
  • Upload date:
  • Size: 4.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for ray_embedding-0.11.1.tar.gz
Algorithm Hash digest
SHA256 e3176f4b138360a9277c7387aab81cbc686a68bbf41a8ce914f2a1ef53483c5d
MD5 9a0036eb54fe2ced5b114a6a49361215
BLAKE2b-256 766509f7131bae724d8d8d3085cf467f23643cf427db9eeac17e2ee4c3b0bc52

See more details on using hashes here.

File details

Details for the file ray_embedding-0.11.1-py3-none-any.whl.

File metadata

  • Download URL: ray_embedding-0.11.1-py3-none-any.whl
  • Upload date:
  • Size: 5.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for ray_embedding-0.11.1-py3-none-any.whl
Algorithm Hash digest
SHA256 4ec13ea14830e17488571e14d0b6a9f4cc9499ba043ad7e83202e9c454161d5c
MD5 bf272754b82d7279925489283cbfa974
BLAKE2b-256 0a51261f1ef19280ac70f5c7b1864792d853691fbc1c71dd59caf060254c82ad

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page