from transformers import LlamaForCausalLM
from .configuration_cognitivess import CognitivessConfig

class CognitivessForCausalLM(LlamaForCausalLM):
    config_class = CognitivessConfig