qherreros's picture
Create README.md
2d024cd verified
|
Raw
History Blame
3.02 kB
metadata
base_model:
  - Qwen/Qwen3-0.6B
license: cc



Jina AI: Your Search Foundation, Supercharged!

jina-embeddings-v5-text-small-classification: Classification-Targeted Embedding Distillation

Blog | Technical Report | API

Model Overview

jina-embeddings-v5-text-small-classification is a compact, high-performance text embedding model designed for classification.

It is part of the jina-embeddings-v5-text model family, which also includes jina-embeddings-v5-text-nano, a smaller model for more resource-constrained use cases.

Trained using a novel approach that combines distillation with task-specific contrastive losses, jina-embeddings-v5-text-small-classification outperforms existing state-of-the-art models of similar size across diverse embedding benchmarks.

Feature Value
Parameters 677M
Supported Tasks classification
Max Sequence Length 32768
Embedding Dimension 1024
Matryoshka Dimensions 32, 64, 128, 256, 512, 768, 1024
Pooling Strategy Last-token pooling
Base Model Qwen/Qwen3-0.6B-Base

Training and Evaluation

For training details and evaluation results, see our technical report.

Usage

Requirements

The following Python packages are required:

  • transformers>=4.57.0
  • torch>=2.8.0
  • peft>=0.15.2

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.
via vLLM
from vllm import LLM

# Initialize model
name = "jinaai/jina-embeddings-v5-text-small-classification"
model = LLM(model=name, task="embed", dtype="bfloat16")

# Create text prompts
document1 = "Overview of climate change impacts on coastal cities"
document1_prompt = f"Document: {document1}"

document2 = "The impacts of climate change on large cities"
document2_prompt = f"Document: {document2}"

# Encode all prompts
prompts = [document1_prompt, document2_prompt]
outputs = model.encode(prompts, pooling_task="embed")


embed_document1 = outputs[0].outputs.data
embed_document2 = outputs[1].outputs.data

License

The model is licensed under CC BY-NC 4.0. For commercial use, please contact us.

Citation

If you find jina-embeddings-v5-text-small-classification useful in your research, please cite the following paper:

[TODO]