cognitivess commited on
Commit
cdcb4e1
1 Parent(s): cab4193

Upload cognitivess_model/__init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. cognitivess_model/__init__.py +7 -0
cognitivess_model/__init__.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ from .configuration_cognitivess import CognitivessConfig
2
+ from .modeling_cognitivess import CognitivessForCausalLM
3
+
4
+ from transformers import AutoConfig, AutoModelForCausalLM
5
+
6
+ AutoConfig.register('cognitivess', CognitivessConfig)
7
+ AutoModelForCausalLM.register(CognitivessConfig, CognitivessForCausalLM)