From cfd5c20286b7f4be705fe976ad38d0307ba8cd6f Mon Sep 17 00:00:00 2001 From: Ziyang Ma Date: Thu, 24 Sep 2026 18:07:37 +0800 Subject: [PATCH] [XPU][UT] Align HF and vLLM inputs for Qwen2 embedding test by preventing Sentence Transformers from applying chat template (#58117) Signed-off-by: RyanMa29 --- requirements/test/xpu.in | 1 + requirements/test/xpu.txt | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/requirements/test/xpu.in b/requirements/test/xpu.in index 796e76eb44..d2ad5a0cfd 100644 --- a/requirements/test/xpu.in +++ b/requirements/test/xpu.in @@ -38,6 +38,7 @@ jiwer bm25s pystemmer mteb[bm25s]>=2, <3 # required for mteb test +sentence-transformers==5.2.0 # required for embedding tests num2words pqdm diff --git a/requirements/test/xpu.txt b/requirements/test/xpu.txt index 8d91ac191c..ed8827f5db 100644 --- a/requirements/test/xpu.txt +++ b/requirements/test/xpu.txt @@ -540,7 +540,6 @@ numpy==2.2.6 # sacrebleu # scikit-learn # scipy - # sentence-transformers # soundfile # soxr # tensorizer @@ -936,8 +935,10 @@ scipy==1.18.0 # pytrec-eval-terrier # scikit-learn # sentence-transformers -sentence-transformers==5.7.0 - # via mteb +sentence-transformers==5.2.0 + # via + # -r requirements/test/xpu.in + # mteb sentencepiece==0.2.1 # via -r requirements/test/../common.txt sentry-sdk==2.68.0 @@ -1047,7 +1048,6 @@ tokenizers==0.23.1 # -c requirements/common.txt # -r requirements/test/../common.txt # cohere - # sentence-transformers # transformers torch==2.14.0+xpu # via