Skip to content

[Core] Move save_tensorized_model logic to Worker#35825

Merged
WoosukKwon merged 1 commit intovllm-project:mainfrom
njhill:save-tensorized
Mar 3, 2026
Merged

[Core] Move save_tensorized_model logic to Worker#35825
WoosukKwon merged 1 commit intovllm-project:mainfrom
njhill:save-tensorized

Conversation

@njhill
Copy link
Copy Markdown
Member

@njhill njhill commented Mar 3, 2026

It's cleaner for this to live in the worker, it should then work for both V1 and V2 model runners.

It's cleaner for this to live in the worker, it should then work for both V1 and V2 model runners.

Signed-off-by: Nick Hill <nickhill123@gmail.com>
@mergify mergify Bot added the v1 label Mar 3, 2026
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request refactors the save_tensorized_model logic by moving it from GPUModelRunner to the Worker class. This change makes the logic more generic and available to both V1 and V2 model runners. The implementation has been moved, and imports have been updated accordingly. The new implementation in the Worker class correctly accesses the model and its configuration. The changes appear correct, and I have not identified any issues.

@njhill njhill added the ready ONLY add when PR is ready to merge/full CI is needed label Mar 3, 2026
@WoosukKwon WoosukKwon merged commit d15c3b9 into vllm-project:main Mar 3, 2026
52 of 54 checks passed
@njhill njhill deleted the save-tensorized branch March 3, 2026 23:33
Copilot AI pushed a commit to machov/vllm that referenced this pull request Mar 10, 2026
Signed-off-by: Nick Hill <nickhill123@gmail.com>
avinashsingh77 pushed a commit to avinashsingh77/vllm that referenced this pull request Mar 12, 2026
Signed-off-by: Nick Hill <nickhill123@gmail.com>
wendyliu235 pushed a commit to wendyliu235/vllm-public that referenced this pull request Mar 18, 2026
Signed-off-by: Nick Hill <nickhill123@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready ONLY add when PR is ready to merge/full CI is needed v1

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants