[Bug] Correct Weight Offloader's Attribute Name for torch.nn.Parameter (#25786)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
co-authored by
github-actions[bot]
parent
b7085d3860
commit
0d3a94f643
@@ -452,7 +452,7 @@ def _move_param_to_meta(module, param_name):
|
||||
data=new_data,
|
||||
requires_grad=False,
|
||||
)
|
||||
if hasattr(old_param, "weihgt_loader"):
|
||||
if hasattr(old_param, "weight_loader"):
|
||||
new_param.weight_loader = old_param.weight_loader
|
||||
else:
|
||||
new_param.weight_loader = lambda *args, **kwargs: None
|
||||
|
||||
Reference in New Issue
Block a user