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

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.11.3.tar.gz
  • Upload date:
  • Size: 4.8 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.3.tar.gz
Algorithm Hash digest
SHA256 68966a6a94f2c8502c5045160bb36d0b25812c92f4e8c366ceb53182a53f091a
MD5 9976b2c41419b6c79269be7c09501c6b
BLAKE2b-256 f219968d9ae107ae6c70b6a2872615104b40189df1d9acf0330a0113dbac5fa6

See more details on using hashes here.

File details

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

File metadata

  • Download URL: ray_embedding-0.11.3-py3-none-any.whl
  • Upload date:
  • Size: 6.1 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.3-py3-none-any.whl
Algorithm Hash digest
SHA256 d788d64121c2081c9efe92600f9a2b39709a44ad606fef75b5cfca3d3ccb30b8
MD5 6a4b31244fc285e4676b4cfc8106565a
BLAKE2b-256 138d444479684a87e32d958e08808dee2d0b585254119c416db13b2991eda775

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