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

Isotr0py · gemini-code-assist[bot] · web-flow · commit 0d21b2664c82 · 2025-07-11T11:21:52.000-07:00
Signed-off-by: Isotr0py &lt;2037008807@qq.com&gt;
Co-authored-by: gemini-code-assist[bot] &lt;176961590+gemini-code-assist[bot]@users.noreply.github.com&gt;
diff --git a/tests/models/language/generation/test_common.py b/tests/models/language/generation/test_common.py
@@ -90,7 +90,7 @@
             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],
         )
     ])

Original file line number	Diff line number	Diff line change
`@@ -90,7 +90,7 @@`
`90`	`90`	`marks=[pytest.mark.core_model],`
`91`	`91`	`),`
`92`	`92`	`pytest.param(`
`93`		`- "Qwen/Qwen1.5-MoE-A2.7B-Chat",`
	`93`	`+ "allenai/OLMoE-1B-7B-0924-Instruct",`
`94`	`94`	`marks=[pytest.mark.cpu_model],`
`95`	`95`	`)`
`96`	`96`	`])`