mct-nightly 2.0.0.20240430.357__py3-none-any.whl → 2.0.0.20240502.415__py3-none-any.whl

{mct_nightly-2.0.0.20240430.357.dist-info → mct_nightly-2.0.0.20240502.415.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 2.0.0.20240430.357
+Version: 2.0.0.20240502.415
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN

{mct_nightly-2.0.0.20240430.357.dist-info → mct_nightly-2.0.0.20240502.415.dist-info}/RECORD

@@ -1,4 +1,4 @@
-model_compression_toolkit/__init__.py,sha256=AWT1k31-QfUwqyEmjAKtkm3O_DfyJZt3JVNIWl0H02Y,1573
+model_compression_toolkit/__init__.py,sha256=v97MrXgZol62-SQfUylkr3kRO-v9bULIikFMXH2u3Yw,1573
 model_compression_toolkit/constants.py,sha256=yIJyJ-e1WrDeKD9kG15qkqfYnoj7J1J2CxnJDt008ik,3756
 model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
 model_compression_toolkit/logger.py,sha256=3DByV41XHRR3kLTJNbpaMmikL8icd9e1N-nkQAY9oDk,4567
@@ -417,7 +417,7 @@ model_compression_toolkit/target_platform_capabilities/immutable.py,sha256=YhROB
 model_compression_toolkit/target_platform_capabilities/target_platform/__init__.py,sha256=bD2xE101rRw7pGarGiBzYraeFF7FUnzyWtAVzwu5JT4,1724
 model_compression_toolkit/target_platform_capabilities/target_platform/current_tp_model.py,sha256=1Glr4qKDJfdk5TwM5fzZ12XzgbpQFioDOxb475905gk,2013
 model_compression_toolkit/target_platform_capabilities/target_platform/fusing.py,sha256=f3xBAI6ivPvEj4lw8cAvTKdIbs7CRdLAa_0LvhGw3Dg,3924
-model_compression_toolkit/target_platform_capabilities/target_platform/op_quantization_config.py,sha256=xMBarITs2g_pCakep1vlNpM9Au5kQubTtFQAFJhvtYs,14248
+model_compression_toolkit/target_platform_capabilities/target_platform/op_quantization_config.py,sha256=kk2c0XjeW3WbK0NihV1j_EffnnpfOXJEI4hJeu6amEc,14693
 model_compression_toolkit/target_platform_capabilities/target_platform/operators.py,sha256=rRmrmPBY4rxCWVpEc6FxeOPUFh8MkfwgQsqD82U9a7w,3108
 model_compression_toolkit/target_platform_capabilities/target_platform/target_platform_model.py,sha256=55EQyP4McJBwsw9_l9eseEZ-V2ygQfdRXjjJLF6HdIM,9482
 model_compression_toolkit/target_platform_capabilities/target_platform/target_platform_model_component.py,sha256=TDbNQwmF7Id-FoIQZlR7ZOcz_nRb4XKBmDihAgKT0u8,1392
@@ -483,8 +483,8 @@ model_compression_toolkit/trainable_infrastructure/keras/quantize_wrapper.py,sha
 model_compression_toolkit/trainable_infrastructure/keras/quantizer_utils.py,sha256=MVwXNymmFRB2NXIBx4e2mdJ1RfoHxRPYRgjb1MQP5kY,1797
 model_compression_toolkit/trainable_infrastructure/pytorch/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
 model_compression_toolkit/trainable_infrastructure/pytorch/base_pytorch_quantizer.py,sha256=7bbzqJN8ZAycVDvZr_5xC-niTAR5df8f03Kooev_pfg,3047
-mct_nightly-2.0.0.20240430.357.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-mct_nightly-2.0.0.20240430.357.dist-info/METADATA,sha256=uY2_yLfendzWlFpSTskBdvOuj1jS0LCfgPaOO9pbK0Y,18795
-mct_nightly-2.0.0.20240430.357.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-mct_nightly-2.0.0.20240430.357.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-2.0.0.20240430.357.dist-info/RECORD,,
+mct_nightly-2.0.0.20240502.415.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+mct_nightly-2.0.0.20240502.415.dist-info/METADATA,sha256=8AajZWyDxrjK7zlV_zeHGx-4wghroX01bu9ufvXTO84,18795
+mct_nightly-2.0.0.20240502.415.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+mct_nightly-2.0.0.20240502.415.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-2.0.0.20240502.415.dist-info/RECORD,,

model_compression_toolkit/__init__.py

@@ -27,4 +27,4 @@ from model_compression_toolkit import data_generation
 from model_compression_toolkit import pruning
 from model_compression_toolkit.trainable_infrastructure.keras.load_model import keras_load_quantized_model
 
-__version__ = "2.0.0.20240430.000357"
+__version__ = "2.0.0.20240502.000415"

model_compression_toolkit/target_platform_capabilities/target_platform/op_quantization_config.py

@@ -299,7 +299,15 @@ class QuantizationConfigOptions(object):
         """
         qc_options = copy.deepcopy(self)
 
-        for qc in qc_options.quantization_config_list:
+        # Extract the list of existing quantization configurations from qc_options
+
+        # Check if the base_config is already included in the quantization configuration list
+        # If not, add base_config to the list of configurations to update
+        cfgs_to_update = [cfg for cfg in qc_options.quantization_config_list]
+        if not any(qc_options.base_config is cfg for cfg in cfgs_to_update):
+            cfgs_to_update.append(qc_options.base_config)
+
+        for qc in cfgs_to_update:
             if layer_attrs_mapping is None:
                 qc.attr_weights_configs_mapping = {}
             else: