Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
71799fd0
Unverified
Commit
71799fd0
authored
Jun 27, 2025
by
Michael Goin
Committed by
GitHub
Jun 27, 2025
Browse files
[CI Failure] Fix OOM with test_oot_registration_embedding (#20144)
Signed-off-by:
mgoin
<
mgoin64@gmail.com
>
parent
e9fd658a
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
tests/models/test_oot_registration.py
tests/models/test_oot_registration.py
+3
-1
No files found.
tests/models/test_oot_registration.py
View file @
71799fd0
...
...
@@ -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
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment