Update cognitivess_model/modeling_cognitivess.py

cognitivess_model/modeling_cognitivess.py

@@ -3,3 +3,8 @@ from .configuration_cognitivess import CognitivessConfig
 
 class CognitivessForCausalLM(LlamaForCausalLM):
     config_class = CognitivessConfig
+
+    def __init__(self, config):
+        super().__init__(config)
+        if config.quantization_config:
+            self.quantize(config.quantization_config)