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)