Skip to content

Commit 7b9282d

Browse files
committed
[ci] Separate vllm tests
1 parent 89403fe commit 7b9282d

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

.github/workflows/integration.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,9 @@ jobs:
157157
- test: TestVllm1
158158
instance: g6
159159
failure-prefix: lmi
160+
- test: TestVllm2
161+
instance: g6
162+
failure-prefix: lmi
160163
- test: TestVllmCustomHandlers
161164
instance: g6
162165
failure-prefix: lmi

tests/integration/tests.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -587,6 +587,11 @@ def test_llama_68m_speculative_medusa(self):
587587
r.launch()
588588
client.run("vllm llama-68m-speculative-medusa".split())
589589

590+
591+
@pytest.mark.vllm
592+
@pytest.mark.gpu_4
593+
class TestVllm2:
594+
590595
def test_llama_68m_speculative_eagle(self):
591596
with Runner('lmi', 'llama-68m-speculative-eagle') as r:
592597
prepare.build_vllm_async_model("llama-68m-speculative-eagle")

0 commit comments

Comments
 (0)