qherreros commited on
Commit
44aa200
·
verified ·
1 Parent(s): 92ae623

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -2
README.md CHANGED
@@ -42,9 +42,10 @@ For training details and evaluation results, see our [technical report](https://
42
 
43
  The following Python packages are required:
44
 
45
- - `transformers>=4.57.0`
46
  - `torch>=2.8.0`
47
  - `peft>=0.15.2`
 
48
 
49
  ### Optional / Recommended
50
  - **flash-attention**: Installing [flash-attention](https://github.com/Dao-AILab/flash-attention) is recommended for improved inference speed and efficiency, but not mandatory.
@@ -60,7 +61,12 @@ from vllm import LLM
60
 
61
  # Initialize model
62
  name = "jinaai/jina-embeddings-v5-text-small-clustering"
63
- model = LLM(model=name, task="embed", dtype="bfloat16")
 
 
 
 
 
64
 
65
  # Create text prompts
66
  document1 = "Overview of climate change impacts on coastal cities"
 
42
 
43
  The following Python packages are required:
44
 
45
+ - `transformers>=5.1.0`
46
  - `torch>=2.8.0`
47
  - `peft>=0.15.2`
48
+ - `vllm>=0.15.1`
49
 
50
  ### Optional / Recommended
51
  - **flash-attention**: Installing [flash-attention](https://github.com/Dao-AILab/flash-attention) is recommended for improved inference speed and efficiency, but not mandatory.
 
61
 
62
  # Initialize model
63
  name = "jinaai/jina-embeddings-v5-text-small-clustering"
64
+ model = LLM(
65
+ model=name,
66
+ dtype="float16",
67
+ runner="pooling",
68
+ pooler_config=PoolerConfig(seq_pooling_type="LAST", normalize=True)
69
+ )
70
 
71
  # Create text prompts
72
  document1 = "Overview of climate change impacts on coastal cities"