Skip to content

Commit cb39526

Browse files
committed
[6/N][CI/UT] Add pooling model test
Signed-off-by: MengqingCao <cmq0113@163.com>
1 parent 96d6fa7 commit cb39526

File tree

9 files changed

+860
-2
lines changed

9 files changed

+860
-2
lines changed

.github/workflows/vllm_ascend_test.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ jobs:
124124
if [[ "${{ matrix.os }}" == "linux-arm64-npu-1" ]]; then
125125
pytest -sv tests/singlecard/test_offline_inference.py
126126
pytest -sv tests/ops
127+
pytest -sv tests/singlecard/embedding
127128
else
128129
pytest -sv tests/multicard/test_offline_inference_distributed.py
129130
pytest -sv tests/ops
@@ -137,6 +138,7 @@ jobs:
137138
if [[ "${{ matrix.os }}" == "linux-arm64-npu-1" ]]; then
138139
pytest -sv tests/singlecard/test_offline_inference.py
139140
pytest -sv tests/ops
141+
pytest -sv tests/singlecard/embedding
140142
else
141143
pytest -sv tests/multicard/test_offline_inference_distributed.py
142144
pytest -sv tests/ops

requirements-dev.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@ pytest >= 6.0
44
pytest-asyncio
55
lm-eval
66
ray
7+
sentence_transformers

0 commit comments

Comments
 (0)