[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:
1 parent
b968373c88
commit
65bc968465
9 files changed
+41
-36
No files matched your search
@@ -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"``).
|
||||
|
||||
Reference in new issue
Block a user