From 71799fd005ca08c9c362e548945a3dde93790fec Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Fri, 27 Jun 2025 12:21:04 +0900 Subject: [PATCH] [CI Failure] Fix OOM with test_oot_registration_embedding (#20144) Signed-off-by: mgoin --- tests/models/test_oot_registration.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/models/test_oot_registration.py b/tests/models/test_oot_registration.py index ef0ad613d5252..59de35644c12d 100644 --- a/tests/models/test_oot_registration.py +++ b/tests/models/test_oot_registration.py @@ -53,7 +53,9 @@ def test_oot_registration_embedding( with monkeypatch.context() as m: m.setenv("VLLM_PLUGINS", "register_dummy_model") prompts = ["Hello, my name is", "The text does not matter"] - llm = LLM(model=dummy_gemma2_embedding_path, load_format="dummy") + llm = LLM(model=dummy_gemma2_embedding_path, + load_format="dummy", + max_model_len=2048) outputs = llm.embed(prompts) for output in outputs: