File size: 301 Bytes
cdcb4e1 |
1 2 3 4 5 6 7 8 |
from .configuration_cognitivess import CognitivessConfig
from .modeling_cognitivess import CognitivessForCausalLM
from transformers import AutoConfig, AutoModelForCausalLM
AutoConfig.register('cognitivess', CognitivessConfig)
AutoModelForCausalLM.register(CognitivessConfig, CognitivessForCausalLM)
|