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.10.15.tar.gz (4.0 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.10.15-py3-none-any.whl (4.8 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.10.15.tar.gz
  • Upload date:
  • Size: 4.0 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.10.15.tar.gz
Algorithm Hash digest
SHA256 b7dfd734fb0d29e4595ff347b61217f8a9d93d478362b2d0e149875d7aa91d58
MD5 cf9eeeacfad77988cbfaadced0ec588d
BLAKE2b-256 426e692e88b5b2b3d296c244d24a96f6367c2490c5e0b26fc36dbc5e67fe9d7e

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for ray_embedding-0.10.15-py3-none-any.whl
Algorithm Hash digest
SHA256 437728b37f26277876cc3dd76a4f07b4bc0b632a3d6c0293da9242a9553c0c19
MD5 bf260519bff2959ee28076a37a3ca5de
BLAKE2b-256 d42bd94b1caf192491c25d6f965497421861cec808e3d4715bc3f81954cbb7bd

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