biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[CI] Replace large models with tiny alternatives in tests (#24057)

Browse Source 
Signed-off-by: Tahsin Tunan <tahsintunan@gmail.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Nick Hill <nhill@redhat.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Tahsin Tunan
2025-10-16 20:51:27 +06:00
committed by GitHub
parent 02d709a6f1
commit 43721bc67f
17 changed files with 118 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/v1/sample/test_sampling_params_e2e.py
View File

@@ -5,16 +5,13 @@ import pytest
from vllm import LLM, SamplingParams
MODEL = "meta-llama/Llama-3.2-1B"
MODEL = "hmellor/tiny-random-LlamaForCausalLM"
PROMPT = "Hello my name is Robert and I"
@pytest.fixture(scope="module")
def llm() -> LLM:
# Disable prefix caching so that we can test prompt logprobs.
# TODO remove this after https://github.com/vllm-project/vllm/pull/13949
# is merged
return LLM(MODEL, enforce_eager=True, enable_prefix_caching=False)
return LLM(MODEL, enforce_eager=True)
def test_n_gt_1(llm):