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

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.13.17.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.13.17.tar.gz
Algorithm Hash digest
SHA256 b8593f0b1a45e23e3f275cfca4ead552a6823ce8a6b1648af1d66fe8a7a5ca50
MD5 18b68185c16266da351ed003bbbc2858
BLAKE2b-256 7de548df03f1b85c81b34f5731803c9ebdc4a3a2f2a15aaa0c312cb0f9bee13b

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for ray_embedding-0.13.17-py3-none-any.whl
Algorithm Hash digest
SHA256 4b360e7932be5ca466804b66f18b49aa169e0597026cf691687bf0476ede7ee5
MD5 320f93097c830c57ac0308bc2ff2f1ec
BLAKE2b-256 015a2a006c697640f532e594c052db6a2002e36c953586ef6a2e56fd5407d389

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