get_fusion_params

torchtune.modules.model_fusion.get_fusion_params(model: Module) → dict[str, torch.nn.parameter.Parameter]

Return the subset of parameters from a model that correspond to fused modules. Assumes that any fusion class has defined the fusion_params() method.

Parameters:

model (nn.Module) – Instance of model class containing some fusion params.

Returns:

the subset of model’s state dict containing

only adapter parameters.

Return type:

dict[str, nn.Parameter]