cognitivess
commited on
Commit
•
cdcb4e1
1
Parent(s):
cab4193
Upload cognitivess_model/__init__.py with huggingface_hub
Browse files
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)
|