def METHOD_NAME(self, enable_training=False):
    qw = self.client_module.attn.q_proj.weight
    kw = self.client_module.attn.k_proj.weight
    vw = self.client_module.attn.v_proj.weight
    qkvw = Parameter(torch.cat((qw, kw, vw), dim=0), requires_grad=enable_training)
    return qkvw, \
           None, \
           self.client_module.attn.out_proj.weight, \
           None,