Update cognitivess_model/configuration_cognitivess.py

cognitivess_model/configuration_cognitivess.py

@@ -1,7 +1,8 @@
 from transformers import LlamaConfig
 
 class CognitivessConfig(LlamaConfig):
-    model_type = 'cognitivess'
+    model_type = "cognitivess"
 
-    def __init__(self, **kwargs):
+    def __init__(self, quantization_config=None, **kwargs):
         super().__init__(**kwargs)
+        self.quantization_config = quantization_config