xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 07:04:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] Fix OOM in language generation test (#20814)

Browse Source 
Signed-off-by: Isotr0py <2037008807@qq.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Isotr0py 2025-07-12 02:21:52 +08:00 committed by GitHub
parent 9907fc4494
commit 0d21b2664c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/language/generation/test_common.py
View File

@ -90,7 +90,7 @@ AITER_MODEL_LIST = [
marks=[pytest.mark.core_model], marks=[pytest.mark.core_model],
), ),
pytest.param( pytest.param(
"Qwen/Qwen1.5-MoE-A2.7B-Chat", "allenai/OLMoE-1B-7B-0924-Instruct",
marks=[pytest.mark.cpu_model], marks=[pytest.mark.cpu_model],
) )
]) ])