Skip to content

Commit 7c03c68

Browse files
[3/N][CI/UT] add spec decode e2e UT
Co-authored-by: mengwei805 <[email protected]> Co-authored-by: MengqingCao <[email protected]> Signed-off-by: mengwei805 <[email protected]>
1 parent edeadde commit 7c03c68

File tree

4 files changed

+826
-2
lines changed

4 files changed

+826
-2
lines changed

.github/workflows/vllm_ascend_test.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ jobs:
148148
- name: Run vllm-project/vllm-ascend key feature test
149149
if: steps.filter.outputs.speculative_tests_changed
150150
run: |
151-
pytest -sv tests/spec_decode
151+
pytest -sv tests/spec_decode/e2e/test_eagle_correctness.py::test_llama2_eagle_e2e_greedy_correctness
152152
153153
- name: Run vllm-project/vllm test
154154
run: |

tests/conftest.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,14 +62,15 @@ def __init__(
6262
enable_chunked_prefill: bool = False,
6363
swap_space: int = 4,
6464
enforce_eager: Optional[bool] = False,
65+
trust_remote_code: Optional[bool] = True,
6566
**kwargs,
6667
) -> None:
6768
self.model = LLM(
6869
model=model_name,
6970
task=task,
7071
tokenizer=tokenizer_name,
7172
tokenizer_mode=tokenizer_mode,
72-
trust_remote_code=True,
73+
trust_remote_code=trust_remote_code,
7374
dtype=dtype,
7475
swap_space=swap_space,
7576
enforce_eager=enforce_eager,

0 commit comments

Comments
 (0)