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