[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>
2025-07-12 02:21:52 +08:00
parent 9907fc4494
commit 0d21b2664c
1 changed files with 1 additions and 1 deletions
--- 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],
        )
    ])