The following Python packages are required:

• transformers>=5.1.0
• torch>=2.8.0
• peft>=0.15.2
• vllm>=0.15.1

Optional / Recommended

• flash-attention: Installing flash-attention is recommended for improved inference speed and efficiency, but not mandatory.
• sentence-transformers: If you want to use the model via the sentence-transformers interface, install this package as well.

The fastest way to use v5-text in production. Elastic Inference Service (EIS) provides managed embedding inference with built-in scaling, so you can generate embeddings directly within your Elastic deployment.

PUT inference/textembedding/jina-v5
{
  "service": "elastic",
  "service_settings": {
    "model_id": "jina-embeddings-v5-text-small"
  }
}

See the Elastic Inference Service documentation for setup details.

from sentence_transformers import SentenceTransformer
import torch

model = SentenceTransformer(
    "jinaai/jina-embeddings-v5-text-small-retrieval",
    model_kwargs={"dtype": torch.bfloat16},  # Recommended for GPUs
    configkwargs={"attnimplementation": "flashattention_2"},  # Recommended but optional
)
Optional: set truncate_dim in encode() to control embedding size


query = "Which planet is known as the Red Planet?"
documents = [
    "Venus is often called Earth's twin because of its similar size and proximity.",
    "Mars, known for its reddish appearance, is often referred to as the Red Planet.",
    "Jupiter, the largest planet in our solar system, has a prominent red spot.",
    "Saturn, famous for its rings, is sometimes mistaken for the Red Planet.",
]

Encode query and documents
queryembeddings = model.encode(sentences=query, promptname="query")
documentembeddings = model.encode(sentences=documents, promptname="document")
print(queryembeddings.shape, documentembeddings.shape)
(1024,) (4, 1024)

similarity = model.similarity(queryembeddings, documentembeddings)
print(similarity)
tensor([[0.4860, 0.7611, 0.5914, 0.6188]])

from vllm import LLM
from vllm.config.pooler import PoolerConfig

Initialize model
name = "jinaai/jina-embeddings-v5-text-small-retrieval"
model = LLM(
    model=name,
    dtype="float16",
    runner="pooling",
    poolerconfig=PoolerConfig(seqpooling_type="LAST", normalize=True),
)

Create text prompts
query = "Overview of climate change impacts on coastal cities"
query_prompt = f"Query: {query}"

document = "The impacts of climate change on coastal cities are significant.."
document_prompt = f"Document: {document}"

Encode all prompts
prompts = [queryprompt, documentprompt]
outputs = model.encode(prompts, pooling_task="embed")

• Via Docker on CPU:

docker run -p 8080:80 \
    ghcr.io/huggingface/text-embeddings-inference:cpu-1.9 \
    --model-id jinaai/jina-embeddings-v5-text-small-retrieval \
    --dtype float32 --pooling last-token

• Via Docker on NVIDIA GPU (Turing, Ampere, Ada Lovelace, Hopper or Blackwell):

docker run --gpus all --shm-size 1g -p 8080:80 \
    ghcr.io/huggingface/text-embeddings-inference:cuda-1.9 \
    --model-id jinaai/jina-embeddings-v5-text-small-retrieval \
    --dtype float16 --pooling last-token

Alternatively, you can also run with cargo, more information can be found in the Text Embeddings Inference documentation.

Send a request to /v1/embeddings to generate embeddings via the OpenAI Embeddings API:

curl -X POST http://127.0.0.1:8080/v1/embeddings \
  -H "Content-Type: application/json" \
  -d '{
    "model": "jinaai/jina-embeddings-v5-text-small-retrieval",
    "input": [
      "Query: Overview of climate change impacts on coastal cities",
      "Document: The impacts of climate change on coastal cities are significant...",
    ]
  }'

Or rather via the Text Embeddings Inference API specification instead, to prevent from manually formatting the inputs:

curl -X POST http://127.0.0.1:8080/embed \
  -H "Content-Type: application/json" \
  -d '{
    "inputs": "Overview of climate change impacts on coastal cities",
    "prompt_name": "query",
  }'

llama-server -hf jinaai/jina-embeddings-v5-text-small-retrieval:F16 --embedding --pooling last -ub 32768

Client:

curl -X POST "http://127.0.0.1:8080/v1/embeddings" \
  -H "Content-Type: application/json" \
  -d '{
    "input": [
      "Query: A beautiful sunset over the beach",
      "Query: Un beau coucher de soleil sur la plage",
      "Document: 海滩上美丽的日落",
      "Document: 浜辺に沈む美しい夕日",
      "Document: Golden sunlight melts into the horizon, painting waves in warm amber and rose, while the sky whispers goodnight to the quiet, endless sea."
    ]
  }'