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
2e5d2137
Unverified
Commit
2e5d2137
authored
Sep 07, 2025
by
Woosuk Kwon
Committed by
GitHub
Sep 07, 2025
Browse files
Skip MM Encoder for non-first PP ranks (#24387)
Signed-off-by:
Woosuk Kwon
<
woosuk.kwon@berkeley.edu
>
parent
0661cb9d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
4 deletions
+1
-4
vllm/v1/worker/gpu_model_runner.py
vllm/v1/worker/gpu_model_runner.py
+1
-4
No files found.
vllm/v1/worker/gpu_model_runner.py
View file @
2e5d2137
...
@@ -1620,14 +1620,11 @@ class GPUModelRunner(LoRAModelRunnerMixin, KVConnectorModelRunnerMixin):
...
@@ -1620,14 +1620,11 @@ class GPUModelRunner(LoRAModelRunnerMixin, KVConnectorModelRunnerMixin):
# _prepare_inputs may reorder the batch, so we must gather multi
# _prepare_inputs may reorder the batch, so we must gather multi
# modal outputs after that to ensure the correct order
# modal outputs after that to ensure the correct order
if
self
.
supports_mm_inputs
:
if
self
.
supports_mm_inputs
and
get_pp_group
().
is_first_rank
:
# Run the multimodal encoder if any.
# Run the multimodal encoder if any.
self
.
_execute_mm_encoder
(
scheduler_output
)
self
.
_execute_mm_encoder
(
scheduler_output
)
mm_embeds
=
self
.
_gather_mm_embeddings
(
scheduler_output
)
mm_embeds
=
self
.
_gather_mm_embeddings
(
scheduler_output
)
else
:
mm_embeds
=
[]
if
self
.
supports_mm_inputs
and
get_pp_group
().
is_first_rank
:
# NOTE(woosuk): To unify token ids and soft tokens (vision
# NOTE(woosuk): To unify token ids and soft tokens (vision
# embeddings), we always use embeddings (rather than token ids)
# embeddings), we always use embeddings (rather than token ids)
# as input to the multimodal model, even when the input is text.
# as input to the multimodal model, even when the input is text.
...
...
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