From 0d21b2664c82448bbea1464496063f9b99cf61ea Mon Sep 17 00:00:00 2001 From: Isotr0py Date: Sat, 12 Jul 2025 02:21:52 +0800 Subject: [PATCH] [Bugfix] Fix OOM in language generation test (#20814) Signed-off-by: Isotr0py <2037008807@qq.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- tests/models/language/generation/test_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/language/generation/test_common.py b/tests/models/language/generation/test_common.py index 8aba68829b10..ea240d227889 100644 --- a/tests/models/language/generation/test_common.py +++ b/tests/models/language/generation/test_common.py @@ -90,7 +90,7 @@ AITER_MODEL_LIST = [ marks=[pytest.mark.core_model], ), pytest.param( - "Qwen/Qwen1.5-MoE-A2.7B-Chat", + "allenai/OLMoE-1B-7B-0924-Instruct", marks=[pytest.mark.cpu_model], ) ])