update

openvinotoolkit · Jan 31, 2025 · 1c51fd8 · 1c51fd8
1 parent 6024dce
commit 1c51fd8
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/nncf/tensorflow/utils/state.py b/nncf/tensorflow/utils/state.py
@@ -15,6 +15,7 @@
 import tensorflow as tf
 
 from nncf.common.compression import BaseCompressionAlgorithmController
+from nncf.tensorflow.quantization.algorithm import TFQuantizationSetup
 
 # TODO(achurkin): remove pylint ignore after 120296 ticked is fixed
 
@@ -105,7 +106,12 @@ def serialize(self) -> str:
 
         :return: A serialized config.
         """
-        return json.dumps(self.config)
+        data = {
+            "quantization": {
+                "quantizer_setup": TFQuantizationSetup.from_state(self.config["quantizer_setup"]).get_state(),
+            }
+        }
+        return json.dumps(data)
 
     def deserialize(self, string_value: str) -> None:
         """