| 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.") | |