[Refactor] Rename SupportsModuleOffload to SupportsComponentDiscovery (#3354)

Signed-off-by: Nick Cao <ncao@redhat.com>
Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
Nick CaoandClaude authored and GitHub committed 2026-05-08 14:16:49 +00:00
1 parent b968373c88
commit 65bc968465
9 files changed
+41 -36

No files matched your search

+6 -2
View File
@@ -59,8 +59,12 @@ class SupportsStepExecution(Protocol):
@runtime_checkable
class SupportsModuleOffload(Protocol):
"""Declares which submodules participate in CPU offload.
class SupportsComponentDiscovery(Protocol):
"""Declares which submodules serve as pipeline components.
Used by the framework to locate DiT, encoder, and VAE modules for
CPU offload, HSDP sharding, and other operations that need to know
the pipeline's internal structure.
All attribute names support dotted paths for nested submodules
(e.g. ``"pipe.transformer"``).