[Refactor] Let diffusion pipelines declare offloadable modules via SupportsModuleOffload#2427

Merged

lishunyang12 merged 10 commits into

vllm-project:mainfrom

NickCao:fix/offload-module-discovery

Apr 22, 2026

Commits on Apr 22, 2026

[Refactor] Let diffusion pipelines declare offloadable modules via SupportsModuleOffload

NickCao
and
claude
committed
[Perf][OmniGen2] Declare offloadable modules for CPU offload discovery

NickCao
and
claude
committed
[Test] Add unit tests for ModuleDiscovery and SupportsModuleOffload

NickCao
and
claude
committed
[Feat] Support dotted paths in SupportsModuleOffload module lists

NickCao
and
claude
committed
[Feat] Add _resident_modules for GPU-pinned modules during layerwise offload

NickCao
and
claude
committed
[Doc] Add SupportsModuleOffload documentation for CPU offload

NickCao
and
claude
committed
[Perf][LTX2] Declare offloadable modules for CPU offload discovery

NickCao
and
claude
committed
[Perf][Bagel] Declare offloadable and resident modules for CPU offload

NickCao
and
claude
committed
[Doc] Update add-diffusion-model skill for SupportsModuleOffload

NickCao
and
claude
committed
[Cleanup] Trim docstrings in SupportsModuleOffload and ModuleDiscovery

NickCao
and
claude
committed