Support S3 Sharded loading with RunAI Model Streamer #16317
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
👋 Hi! Thank you for contributing to the vLLM project. 💬 Join our developer Slack at https://slack.vllm.ai to discuss your PR in #pr-reviews, coordinate on features in #feat- channels, or join special interest groups in #sig- channels. Just a reminder: PRs would not trigger full CI run by default. Instead, it would only run Once the PR is approved and ready to go, your PR reviewer(s) can run CI to test the changes comprehensively before merging. To run CI, PR reviewers can either: Add 🚀 |
omer-dayan
force-pushed
the
omer/runai-model-streamer-sharded
branch
from
87ee684 to
31b91c7
Compare
omer-dayan
changed the title
Signed-off-by: Omer Dayan (SW-GPU) <[email protected]>
omer-dayan
force-pushed
the
omer/runai-model-streamer-sharded
branch
from
31b91c7 to
c447ec3
Compare
1 task
Co-authored-by: Cyrus Leung <[email protected]> Signed-off-by: Omer Dayan (SW-GPU) <[email protected]>
omer-dayan
force-pushed
the
omer/runai-model-streamer-sharded
branch
from
cddf0db to
78264e4
Compare
DarkLight1337
added
the
ready
|
I can't seem to get docker build to pass, can you merge from main again? |
DarkLight1337
approved these changes
Apr 22, 2025
dtransposed
pushed a commit
to dtransposed/vllm
that referenced
this pull request
Apr 22, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]> Signed-off-by: dtransposed <damian@damian-ml-machine.europe-west3-b.c.jetbrains-grazie.internal>
frieda-huang
pushed a commit
to frieda-huang/vllm
that referenced
this pull request
Apr 23, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]> Signed-off-by: Frieda (Jingying) Huang <[email protected]>
liuzijing2014
pushed a commit
to liuzijing2014/vllm
that referenced
this pull request
Apr 25, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]> Signed-off-by: Zijing Liu <[email protected]>
liuzijing2014
pushed a commit
to liuzijing2014/vllm
that referenced
this pull request
Apr 25, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]>
wuisawesome
pushed a commit
to character-tech/vllm
that referenced
this pull request
Apr 28, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]>
jikunshang
pushed a commit
to jikunshang/vllm
that referenced
this pull request
Apr 29, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]>
lk-chen
pushed a commit
to lk-chen/vllm
that referenced
this pull request
Apr 29, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]>
adobrzyn
pushed a commit
to HabanaAI/vllm-fork
that referenced
this pull request
Apr 30, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]> Signed-off-by: Agata Dobrzyniewicz <[email protected]>
RichardoMrMu
pushed a commit
to RichardoMrMu/vllm
that referenced
this pull request
May 12, 2025
) Signed-off-by: Omer Dayan (SW-GPU) <[email protected]> Co-authored-by: Cyrus Leung <[email protected]> Signed-off-by: Mu Huai <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
In this PR we add support for using RunAI Model Streamer with sharded model files.
It allows the user to upload the sharded model files to S3:
and run this model using the following command:
vllm serve s3://core-llm/Llama-3-8b-4/ --tensor-parallel-size 4 --load-format runai_streamer_sharded