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

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.14.3.tar.gz
  • Upload date:
  • Size: 8.5 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.3.tar.gz
Algorithm Hash digest
SHA256 132712b766d9ee88145d721717bb5fe2fc6abc0a6bf5bedc6ed62adc54a0adfc
MD5 3c2ce798f98df4c23561998fcd521173
BLAKE2b-256 c6dc02d3b0d4d31b6b4839d4299139c78a4fac76eee9d8d1dd1744c38d118f06

See more details on using hashes here.

File details

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

File metadata

  • Download URL: ray_embedding-0.14.3-py3-none-any.whl
  • Upload date:
  • Size: 10.3 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.3-py3-none-any.whl
Algorithm Hash digest
SHA256 2e7e0fafa273243061c536fbbec155f1a83c18c8bf6e616d04aa68e0c0c4f3d5
MD5 8f48ae7e681c662a1275cd2e6af09a4f
BLAKE2b-256 d9c53eb2cbffd380c7520af04dd12bb2f85d039fa88c7b2a4336178f6e1a5966

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