cognitivess_model/__init__.py

from .configuration_cognitivess import CognitivessConfig
from .modeling_cognitivess import CognitivessForCausalLM

from transformers import AutoConfig, AutoModelForCausalLM

AutoConfig.register('cognitivess', CognitivessConfig)
AutoModelForCausalLM.register(CognitivessConfig, CognitivessForCausalLM)