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.14.7.tar.gz (8.6 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.14.7-py3-none-any.whl (10.5 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.14.7.tar.gz
  • Upload date:
  • Size: 8.6 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.14.7.tar.gz
Algorithm Hash digest
SHA256 604128fd86864d377874fee1876d18bf43d832100cc00dae1ec12cfd8e385c30
MD5 636d88cbbc4d274fc7b736a0b37c0c94
BLAKE2b-256 d4704828641b59da599c9db91df1752fcf6ef954475777f9f0c7c0d18084d3e7

See more details on using hashes here.

File details

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

File metadata

  • Download URL: ray_embedding-0.14.7-py3-none-any.whl
  • Upload date:
  • Size: 10.5 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.14.7-py3-none-any.whl
Algorithm Hash digest
SHA256 9e738aa062cebcbedfd2d6a0163dcb427fafe78ac22d11d6e259d767aa782f64
MD5 ab397cc95ed21bcb1610ed8f69c37ae7
BLAKE2b-256 ed531ef3354690c832df351f75f8e64dc53354fc04d28671f5b4b2f0eaa638a2

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