from transformers import LlamaConfig class CognitivessConfig(LlamaConfig): model_type = 'cognitivess' def __init__(self, **kwargs): super().__init__(**kwargs)