class CognitivessForCausalLM(LlamaForCausalLM):
    config_class = CognitivessConfig