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

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.10.12.tar.gz
  • Upload date:
  • Size: 3.9 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.10.12.tar.gz
Algorithm Hash digest
SHA256 bfcf65bcd486416dd3a27af288bae6464a96d2a4a3a1eaa74645a4212a4efe54
MD5 1260eb17c2711a0b67baca709f4e7abc
BLAKE2b-256 0ca346d92483564fa13b22b42fc99138d0b49ae4ec56443de30dd4be470c0ac7

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for ray_embedding-0.10.12-py3-none-any.whl
Algorithm Hash digest
SHA256 5a01d3972e7fbb60fd5e041205bed20c97a2ecb4817ac9af9fb35dda69f6b3ce
MD5 d60ffa798b7582237faf88bcf5caf85b
BLAKE2b-256 c44b0eaa3613ab82a24b39834692ae60624a7668b1b6f7ecbe68b9a8a776d959

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