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

Signed-off-by: shen-shanshan <467638484@qq.com>
This commit is contained in:
Shanshan Shen
2026-03-05 06:25:56 +08:00
committed by GitHub
parent d7adcadb9b
commit 562339abc3

View File

@@ -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(
param: Parameter,
shard_size: int,