cognitivess commited on
Commit
78fcf7e
1 Parent(s): 9594be2

Update cognitivess_model/configuration_cognitivess.py

Browse files
cognitivess_model/configuration_cognitivess.py CHANGED
@@ -10,5 +10,5 @@ class CognitivessConfig(LlamaConfig):
10
  def to_dict(self):
11
  output = super().to_dict()
12
  if self.quantization_config is not None:
13
- output["quantization_config"] = self.quantization_config.to_dict()
14
  return output
 
10
  def to_dict(self):
11
  output = super().to_dict()
12
  if self.quantization_config is not None:
13
+ output["quantization_config"] = self.quantization_config.to_dict() if hasattr(self.quantization_config, 'to_dict') else self.quantization_config
14
  return output