Skip to content

[Refactor] Rename offload_model to set_onload_device#643

Merged
kylesayrs merged 4 commits intovllm-project:mainfrom
nrmlthms:rename-offload-model-set-onload-device
Mar 25, 2026
Merged

[Refactor] Rename offload_model to set_onload_device#643
kylesayrs merged 4 commits intovllm-project:mainfrom
nrmlthms:rename-offload-model-set-onload-device

Conversation

@nrmlthms
Copy link
Copy Markdown
Contributor

Summary

  • Added set_onload_device as the canonical replacement for
    offload_model
  • Deprecated offload_model using the existing @deprecated
    decorator, pointing to set_onload_device
  • Removed the offload_device parameter from
    set_onload_device (was already silently ignored in
    offload_model)
  • Updated offloaded_dispatch in utils/offload.py to call
    set_onload_device
  • Updated all tests to use set_onload_device

Part of vllm-project/llm-compressor#2483

Companion PR

vllm-project/llm-compressor#2501

  set_onload_device

  - Add set_onload_device as the canonical function (replaces
  offload_model)
  - Deprecate offload_model using @deprecated decorator
  pointing to set_onload_device
  - Remove offload_device param from set_onload_device (was
  already ignored with warning)
  - Update all internal usages and tests

  Part of vllm-project/llm-compressor#2483
@nrmlthms nrmlthms changed the title Refactor] Rename offload_model to set_onload_device [Refactor] Rename offload_model to set_onload_device Mar 23, 2026
Copy link
Copy Markdown
Collaborator

@brian-dellabetta brian-dellabetta left a comment

Choose a reason for hiding this comment

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

Thanks for the contribution! I see some references to offload_model in README in CT

Copy link
Copy Markdown
Collaborator

@brian-dellabetta brian-dellabetta left a comment

Choose a reason for hiding this comment

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

Thanks for updating the READMEs

Copy link
Copy Markdown
Collaborator

@kylesayrs kylesayrs left a comment

Choose a reason for hiding this comment

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

Really awesome contribution, thanks!

@kylesayrs kylesayrs enabled auto-merge (squash) March 23, 2026 20:38
@mergify
Copy link
Copy Markdown

mergify bot commented Mar 23, 2026

The quality checks have failed. Please run make style and make quality under
the root directory to adddress the lint failures. You will need to install the
dev optional install to get the required linting packages.

@kylesayrs
Copy link
Copy Markdown
Collaborator

Hi @nrmlthms, please run make style; make quality

auto-merge was automatically disabled March 24, 2026 17:16

Head branch was pushed to by a user without write access

@mergify mergify bot removed the quality-failed label Mar 24, 2026
@kylesayrs kylesayrs enabled auto-merge (squash) March 24, 2026 18:02
@kylesayrs kylesayrs merged commit 68f1a7d into vllm-project:main Mar 25, 2026
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants