cognitivess
commited on
Commit
•
6be3d45
1
Parent(s):
665539e
Update cognitivess_model/configuration_cognitivess.py
Browse files
cognitivess_model/configuration_cognitivess.py
CHANGED
@@ -46,3 +46,8 @@ class CognitivessConfig(PretrainedConfig):
|
|
46 |
**kwargs
|
47 |
)
|
48 |
self.attention_dropout = attention_dropout
|
|
|
|
|
|
|
|
|
|
|
|
46 |
**kwargs
|
47 |
)
|
48 |
self.attention_dropout = attention_dropout
|
49 |
+
|
50 |
+
# Register the configuration with the Transformers library
|
51 |
+
from transformers import CONFIG_MAPPING
|
52 |
+
|
53 |
+
CONFIG_MAPPING.register(CognitivessConfig.model_type, CognitivessConfig)
|