Function load_weight

lmdeploy/pytorch/weight_loader/model_weight_loader.py:19–25

Load weight.

(param: torch.nn.Parameter, loaded_weight: torch.Tensor, **kwargs)

def load_weight(param: torch.nn.Parameter, loaded_weight: torch.Tensor, **kwargs):
    """Load weight."""
    if hasattr(param, 'weight_loader'):
        param.weight_loader(param, loaded_weight, **kwargs)
    else:
        assert len(kwargs) == 0
        default_weight_loader(param, loaded_weight)


def default_weight_loader(param: torch.nn.Parameter, loaded_weight: torch.Tensor):