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

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.12.9.tar.gz (7.4 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.12.9-py3-none-any.whl (9.0 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.12.9.tar.gz
  • Upload date:
  • Size: 7.4 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.12.9.tar.gz
Algorithm Hash digest
SHA256 d377fa7a4de96b13fa4e907e5588357338e5b1e52f3348fec99e88daa101a240
MD5 cd5d43530f21bd0a5e6ccc4986be72cc
BLAKE2b-256 f2f352e75faa0382bdc6cc8ba8ff9786e719839113e8dd09e9fd0734e43cf473

See more details on using hashes here.

File details

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

File metadata

  • Download URL: ray_embedding-0.12.9-py3-none-any.whl
  • Upload date:
  • Size: 9.0 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.12.9-py3-none-any.whl
Algorithm Hash digest
SHA256 1b759be6fe2370d106f358803cf1a8e6324ef5e10022425e6d79b8ff42705bf0
MD5 6d6b25bafefe0fb82b6964c5c71cde5e
BLAKE2b-256 f1912fb83c66da18ea604b2d0856e3aa107ae3de96b88b60b41078c69c64354d

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