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
|