def METHOD_NAME(self):
    return self.client_module.post_attention_layernorm.weight, \
           self.client_module.post_attention_layernorm.bias, \
           self.client_module.input_layernorm.weight, \
           self.client_module.input_layernorm.bias