Unverified Commit 562339ab authored by Shanshan Shen's avatar Shanshan Shen Committed by GitHub
Browse files

[Misc] Support OOT linear method registering (#35981)


Signed-off-by: default avatarshen-shanshan <467638484@qq.com>
parent d7adcadb
...@@ -64,6 +64,12 @@ WEIGHT_LOADER_V2_SUPPORTED = [ ...@@ -64,6 +64,12 @@ WEIGHT_LOADER_V2_SUPPORTED = [
] ]
def register_weight_loader_v2_supported_method(cls):
"""Decorator to register a LinearMethod as supporting weight_loader_v2."""
WEIGHT_LOADER_V2_SUPPORTED.append(cls.__name__)
return cls
def adjust_marlin_shard( def adjust_marlin_shard(
param: Parameter, param: Parameter,
shard_size: int, shard_size: int,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment