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.8.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.8-py3-none-any.whl (6.2 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.11.8.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.8.tar.gz
Algorithm Hash digest
SHA256 86b5549b5e5ee20e2753f13d315625eb4f6acbe9263ecbc4550cf7ad11acad2f
MD5 d8c0ce6e1d648ac54fbefd313abc1211
BLAKE2b-256 bbeb1e3bc8e58025cb76b2aad13590ec8e82a7315ffba0e6e1e3b4fed69b99f3

See more details on using hashes here.

File details

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

File metadata

  • Download URL: ray_embedding-0.11.8-py3-none-any.whl
  • Upload date:
  • Size: 6.2 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.8-py3-none-any.whl
Algorithm Hash digest
SHA256 044f12e86923c917d56e00bf9c0a9a9d7855b730a278f9d29a9f27db5b9ee6a8
MD5 6d61a2fc8f559618e26d4883cda21f85
BLAKE2b-256 027187b461fd12d2fbafce68dd1400df03b2a86a18aac9df8ab77d74cd3dc846

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