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
3b1d9c31
Unverified
Commit
3b1d9c31
authored
Apr 09, 2026
by
Cyrus Leung
Committed by
GitHub
Apr 09, 2026
Browse files
[CI/Build] Fix memory cleanup in MM test (#39411)
Signed-off-by:
DarkLight1337
<
tlleungac@connect.ust.hk
>
parent
54d244f2
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
2 deletions
+3
-2
.buildkite/test_areas/models_multimodal.yaml
.buildkite/test_areas/models_multimodal.yaml
+2
-1
tests/models/multimodal/generation/test_memory_leak.py
tests/models/multimodal/generation/test_memory_leak.py
+1
-1
No files found.
.buildkite/test_areas/models_multimodal.yaml
View file @
3b1d9c31
...
...
@@ -56,7 +56,8 @@ steps:
-
tests/models/multimodal
commands
:
-
pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
-
pytest -v -s models/multimodal -m core_model --ignore models/multimodal/generation/test_common.py --ignore models/multimodal/generation/test_ultravox.py --ignore models/multimodal/generation/test_qwen2_5_vl.py --ignore models/multimodal/generation/test_qwen2_vl.py --ignore models/multimodal/generation/test_whisper.py --ignore models/multimodal/processing
-
pytest -v -s models/multimodal -m core_model --ignore models/multimodal/generation/test_common.py --ignore models/multimodal/generation/test_ultravox.py --ignore models/multimodal/generation/test_qwen2_5_vl.py --ignore models/multimodal/generation/test_qwen2_vl.py --ignore models/multimodal/generation/test_whisper.py --ignore models/multimodal/generation/test_memory_leak.py --ignore models/multimodal/processing
-
pytest models/multimodal/generation/test_memory_leak.py -m core_model
-
cd .. && VLLM_WORKER_MULTIPROC_METHOD=spawn pytest -v -s tests/models/multimodal/generation/test_whisper.py -m core_model
# Otherwise, mp_method="spawn" doesn't work
mirror
:
amd
:
...
...
tests/models/multimodal/generation/test_memory_leak.py
View file @
3b1d9c31
...
...
@@ -124,7 +124,7 @@ def llm(monkeypatch):
@
pytest
.
mark
.
core_model
@
pytest
.
mark
.
parametrize
(
"image_urls"
,
[
TEST_IMAGE_NAMES
],
indirect
=
True
)
def
test_
qwen3_vl_
no_memory_leak
(
llm
,
image_urls
:
list
[
str
])
->
None
:
def
test_no_memory_leak
(
llm
,
image_urls
:
list
[
str
])
->
None
:
model_info
=
HF_EXAMPLE_MODELS
.
find_hf_info
(
MODEL_NAME
)
model_info
.
check_available_online
(
on_fail
=
"skip"
)
model_info
.
check_transformers_version
(
on_fail
=
"skip"
)
...
...
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