def METHOD_NAME(self):
    if MegatronLayerPolicy.version == 0:
        return self.client_module.attention.query_key_value.weight.shape[1], \
                self.client_module.attention.num_attention_heads, \
                self.client_module.input_layernorm.eps, \
                DEFAULT_INTERMEDIATE_SIZE
    else:
        return self.client_module.self_attention.query_key_value.weight.shape[1], \
                self.client_module.self_attention.num_attention_heads, \
                self.client_module.input_layernorm.eps, \
                DEFAULT_INTERMEDIATE_SIZE