cognitivess_model/modeling_cognitivess.py

from transformers import LlamaForCausalLM
from .configuration_cognitivess import CognitivessConfig

class CognitivessForCausalLM(LlamaForCausalLM):
    config_class = CognitivessConfig