Maximilien de Bayser ff365eea94 Support bge-m3 sparse embeddings and colbert embeddings (#14526) ...

Signed-off-by: Max de Bayser <mbayser@br.ibm.com>
Signed-off-by: Max de Bayser <maxdebayser@gmail.com>

2026-01-22 23:52:57 +08:00

..

__init__.py

[CI/Build] Reorganize models tests (#17459)

2025-04-30 23:03:08 -07:00

conftest.py

[ROCm][CI] Fix ModernBERT token classification test numerical accuracy on ROCm (#31820)

2026-01-06 23:21:15 +00:00

embed_utils.py

Support bge-m3 sparse embeddings and colbert embeddings (#14526)

2026-01-22 23:52:57 +08:00

test_all_pooling_plus_chunked_prefill.py

[Model][6/N] Improve all pooling task | Support chunked prefill with ALL pooling (#27145)

2025-12-04 13:44:15 +00:00

test_auto_prefix_cache_support.py

Improve enable chunked_prefill & prefix_caching logic. (#26623)

2025-11-27 22:05:48 -08:00

test_bge_m3.py

Support bge-m3 sparse embeddings and colbert embeddings (#14526)

2026-01-22 23:52:57 +08:00

test_classification.py

VLLM_USE_TRITON_FLASH_ATTN V0 variable deprecation (#27611)

2025-11-11 18:34:36 -08:00

test_embedding.py

[Refactor] Separate sequence and token pooling types (#32026)

2026-01-10 04:53:24 +00:00

test_extract_hidden_states.py

[Model][6/N] Improve all pooling task | Support chunked prefill with ALL pooling (#27145)

2025-12-04 13:44:15 +00:00

test_gritlm.py

Support bge-m3 sparse embeddings and colbert embeddings (#14526)

2026-01-22 23:52:57 +08:00

test_head_dtype.py

[Model][0/N] Improve all pooling task | clean up (#25817)

2025-10-13 16:44:50 +08:00

test_mm_classifier_conversion.py

[Refactor] Separate sequence and token pooling types (#32026)

2026-01-10 04:53:24 +00:00

test_multi_vector_retrieval.py

[Model][2/N] Improve all pooling task | Support multi-vector retrieval (#25370)

2025-10-15 11:14:41 +00:00

test_multilabel_classification_support.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_nomic_max_model_len.py

Update rope_scaling to rope_parameters in preparation for Transformers v5 (#28542)

2025-11-19 09:06:36 -08:00

test_pooler_config_init_behaviour.py

[Bugfix] Replace PoolingParams.normalize with use_activation (#32243)

2026-01-13 10:45:42 +00:00

test_reward.py

[CI/Build] Revive skipped reward models e2e test (#31665)

2026-01-05 02:33:46 +00:00

test_scoring.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_splade_sparse_pooler.py

[Model][0/N] Improve all pooling task | clean up (#25817)

2025-10-13 16:44:50 +08:00

test_token_classification.py

[Performance] Improve Triton prefill attention kernel's performance (#32403)

2026-01-17 20:19:59 -08:00

test_truncation_control.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00