4 people 3b6c96a101 [Bugfix][Pooling] Fix wrong scores for chunked prefill under torch.compile (#48901) ...

Signed-off-by: seewoo <seewoo@ucsc.edu>
Signed-off-by: wang.yuqi <yuqi.wang@daocloud.io>
Co-authored-by: Claude <noreply@anthropic.com>
Co-authored-by: wang.yuqi <yuqi.wang@daocloud.io>

2026-07-17 05:19:12 +00: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

[Bugfix][Pooling] Fix wrong scores for chunked prefill under torch.compile (#48901)

2026-07-17 05:19:12 +00: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

[Model] Remove obsolete ERNIE models (#45127)

2026-06-10 20:54:30 +08:00

test_colbert.py

[ROCm][CI] Stabilizing teardown and timeout of flaky tests to prevent rare OOMs (#44761)

2026-06-08 14:11:17 +08:00

test_embedding.py

[Bugfix] Fix normalize still being passed to PoolerConfig (#33794)

2026-02-04 23:56:02 +08: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

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

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

test_splade_sparse_pooler.py

[CI] Fix SPLADE pooler test broken by #38139 (#38495)

2026-03-30 07:48:33 +00: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

[Frontend] Use new Renderer for Completions and Tokenize API (#32863)

2026-01-31 04:51:15 -08:00