from .configuration_cognitivess import CognitivessConfig
from .modeling_cognitivess import CognitivessForCausalLM
from .tokenization_cognitivess import CognitivessTokenizer
from .tokenization_cognitivess_fast import CognitivessTokenizerFast

from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer

print("Registering CognitivessConfig...")
AutoConfig.register("cognitivess", CognitivessConfig)
print("CognitivessConfig registered successfully.")

print("Registering CognitivessForCausalLM...")
AutoModelForCausalLM.register(CognitivessConfig, CognitivessForCausalLM)
print("CognitivessForCausalLM registered successfully.")

print("Registering CognitivessTokenizer and CognitivessTokenizerFast...")
AutoTokenizer.register(CognitivessConfig, 
                       slow_tokenizer_class=CognitivessTokenizer, 
                       fast_tokenizer_class=CognitivessTokenizerFast)
print("CognitivessTokenizer and CognitivessTokenizerFast registered successfully.")