Skip to content

Commit 8e4e791

Browse files
authored
[CI] Add deepseek-v2-lite test (#631)
### What this PR does / why we need it? Add deepseek-v2-lite test, part of #499 --------- Signed-off-by: wangli <wangli858794774@gmail.com>
1 parent cdece86 commit 8e4e791

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

.github/workflows/vllm_ascend_test.yaml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ jobs:
115115
pytest -sv tests/ops
116116
pytest -sv tests/compile
117117
else
118-
pytest -sv tests/multicard/test_offline_inference_distributed.py
118+
pytest -sv -k "QwQ" tests/multicard/test_offline_inference_distributed.py
119119
pytest -sv tests/ops
120120
pytest -sv tests/compile
121121
fi
@@ -128,7 +128,8 @@ jobs:
128128
pytest -sv tests/singlecard/test_offline_inference.py
129129
pytest -sv tests/ops
130130
else
131-
pytest -sv tests/multicard/test_offline_inference_distributed.py
131+
pytest -sv -k "QwQ" tests/multicard/test_offline_inference_distributed.py
132+
pytest -sv -k "DeepSeek" tests/multicard/test_offline_inference_distributed.py
132133
pytest -sv tests/ops
133134
fi
134135

tests/multicard/test_offline_inference_distributed.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,12 @@
2828
from tests.conftest import VllmRunner
2929

3030
os.environ["PYTORCH_NPU_ALLOC_CONF"] = "max_split_size_mb:256"
31+
os.environ["VLLM_USE_MODELSCOPE"] = "True"
3132

3233

3334
@pytest.mark.parametrize("model, distributed_executor_backend", [
3435
("Qwen/QwQ-32B", "mp"),
36+
("deepseek-ai/DeepSeek-V2-Lite", "mp"),
3537
])
3638
def test_models_distributed(model: str,
3739
distributed_executor_backend: str) -> None:

0 commit comments

Comments
 (0)