Taneem IbrahimandGitHub 43eaefba5a [ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791) ...

Signed-off-by: Taneem Ibrahim <taneem.ibrahim@gmail.com>

2026-07-29 06:30:45 -07: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

[ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791)

2026-07-29 06:30:45 -07:00

test_auto_prefix_cache_support.py

[Refactor] Pass Renderer to Input Processor (#34329)

2026-02-11 19:38:11 -08:00

test_bge_m3.py

[Frontend] Remove frontend pooling multi task support. (#37861)

2026-04-21 12:27:44 +00:00

test_classification.py

[ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791)

2026-07-29 06:30:45 -07:00

test_colbert.py

Make bare hugging_face imports forbidden (#49726)

2026-07-25 04:18:39 -07:00

test_embedding.py

[ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791)

2026-07-29 06:30:45 -07:00

test_extract_hidden_states.py

[Frontend] Remove frontend pooling multi task support. (#37861)

2026-04-21 12:27:44 +00:00

test_gritlm.py

[CI] Torch 2.11 flaky test_spec_decode_logprobs and gritlm tests (#45772)

2026-06-23 01:26:54 +00:00

test_head_dtype.py

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

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

test_jina_reranker_v3.py

[Bugfix][Pooling] Forward instruction to Jina reranker scoring prompts (#47590)

2026-07-05 05:39:13 +00:00

test_max_tokens_per_doc.py

[ROCm][CI] Stabilize ROCm pooling and multimodal CI (#42909)

2026-05-18 03:57:59 +00:00

test_mm_classifier_conversion.py

[CI Failure] Fix Language Models Test (Extended Pooling) daily CI Failure (#36761)

2026-03-12 12:22:05 +08:00

test_multi_vector_retrieval.py

[Frontend] Overlap preprocessing and computation for pooling models offline inference (#47699)

2026-07-16 14:00:20 +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

Fix error in Dynamic NTK scaling (#41277)

2026-05-19 17:27:54 -04:00

test_pooler_config_init_behaviour.py

Deprecations for v0.23 and v0.24 (#44992)

2026-06-11 14:35:38 +00:00

test_reward.py

[copy of #45208] CuMem slept-L1 fragmentation accounting (#49208)

2026-07-20 23:08:11 +00:00

test_splade_sparse_pooler.py

[ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791)

2026-07-29 06:30:45 -07:00

test_token_classification.py

[ROCm][CI] Set "highest" matmul precision for reference hf_runner in test_bert_for_masked_lm (#48784)

2026-07-16 05:49:15 +00:00

test_truncation_control.py

[CI][Test] Fix pooling truncation test after VLLMError hierarchy change (#50241)

2026-07-29 12:15:56 +00:00