def METHOD_NAME(self, enable_training=False):
    return self.client_module.self_attention.query_key_value.weight, \
            self.client_module.self_attention.query_key_value.bias, \
            self.client_module.self_attention.dense.weight, \
            self.client_module.self_attention.dense.bias,