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

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.12.10.tar.gz (7.4 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.12.10-py3-none-any.whl (9.0 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_embedding-0.12.10.tar.gz
  • Upload date:
  • Size: 7.4 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.12.10.tar.gz
Algorithm Hash digest
SHA256 a037c2b29fab0d1af63b8a37d5b8e5445a4fb818e642101349ca3258f34e0193
MD5 9e7004d0c060d06a63fc9c23dedd8b68
BLAKE2b-256 18c864a2f571f428b36978ebafffe927e1335ee64470bff4b20b8e6cfc7b6496

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for ray_embedding-0.12.10-py3-none-any.whl
Algorithm Hash digest
SHA256 7eadcdf00da3e3a3af304d31587780c13d193b9d215f118dfa7e3f24cd7ff25b
MD5 fc865fbc13d6333b846ea91026d402b8
BLAKE2b-256 6e1902fb5a9c622bfb2a0cfdcab38fa93597eb365f301ebb646b10caa07e9033

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