Update cognitivess_model/modeling_cognitivess.py

cognitivess_model/modeling_cognitivess.py

@@ -1,10 +1,2 @@
-from transformers import LlamaForCausalLM
-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)
+    config_class = CognitivessConfig