mct-nightly 2.3.0.20250521.607__py3-none-any.whl → 2.3.0.20250523.601__py3-none-any.whl

{mct_nightly-2.3.0.20250521.607.dist-info → mct_nightly-2.3.0.20250523.601.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mct-nightly
-Version: 2.3.0.20250521.607
+Version: 2.3.0.20250523.601
 Summary: A Model Compression Toolkit for neural networks
 Author-email: ssi-dnn-dev@sony.com
 Classifier: Programming Language :: Python :: 3

{mct_nightly-2.3.0.20250521.607.dist-info → mct_nightly-2.3.0.20250523.601.dist-info}/RECORD

@@ -1,5 +1,5 @@
-mct_nightly-2.3.0.20250521.607.dist-info/licenses/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-model_compression_toolkit/__init__.py,sha256=cuGbLTJRbGxD0Hiqw5hjdm1dyBHHuqccQn14eB14CsU,1557
+mct_nightly-2.3.0.20250523.601.dist-info/licenses/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+model_compression_toolkit/__init__.py,sha256=kbat4rP9HQwW-TV_lhwH8T2LSub816Le2SOMWSMjsW8,1557
 model_compression_toolkit/constants.py,sha256=KNgiNLpsMgSYyXMNEbHXd4bFNerQc1D6HH3vpbUq_Gs,4086
 model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
 model_compression_toolkit/logger.py,sha256=L3q7tn3Uht0i_7phnlOWMR2Te2zvzrt2HOz9vYEInts,4529
@@ -333,11 +333,11 @@ model_compression_toolkit/exporter/model_exporter/keras/int8_tflite_exporter.py,
 model_compression_toolkit/exporter/model_exporter/keras/keras_export_facade.py,sha256=NzcX7rxLk__Kpuim_VXaOHS4tyiRtRBoERPE00GbdfA,5862
 model_compression_toolkit/exporter/model_exporter/keras/mctq_keras_exporter.py,sha256=qXXkv3X_wb7t622EOHwXIxfGLGaDqh0T0y4UxREi4Bo,1976
 model_compression_toolkit/exporter/model_exporter/pytorch/__init__.py,sha256=uZ2RigbY9O2PJ0Il8wPpS_s7frgg9WUGd_SHeKGyl1A,699
-model_compression_toolkit/exporter/model_exporter/pytorch/base_pytorch_exporter.py,sha256=UPVkEUQCMZ4Lld6CRnEOPEmlfe5vcQZG0Q3FwRBodD4,4021
+model_compression_toolkit/exporter/model_exporter/pytorch/base_pytorch_exporter.py,sha256=kmddLPXN47b4UOlip8u7zfILcw7rvKHp5UZ_-nII_9U,5369
 model_compression_toolkit/exporter/model_exporter/pytorch/export_serialization_format.py,sha256=bPevy6OBqng41PqytBR55e6cBEuyrUS0H8dWX4zgjQ4,967
 model_compression_toolkit/exporter/model_exporter/pytorch/fakely_quant_onnx_pytorch_exporter.py,sha256=v6DwLNZX5RcWHUfYXfJZXNFXPfHsMm5Wl0N29o0diSc,8856
 model_compression_toolkit/exporter/model_exporter/pytorch/fakely_quant_torchscript_pytorch_exporter.py,sha256=ksWV2A-Njo-wAxQ_Ye2sLIZXBWJ_WNyjT7-qFFwvV2o,2897
-model_compression_toolkit/exporter/model_exporter/pytorch/pytorch_export_facade.py,sha256=I2YZOwzTMLcclGJoruFrHJlOt5mepodX7mGw_GbnqxA,6372
+model_compression_toolkit/exporter/model_exporter/pytorch/pytorch_export_facade.py,sha256=7xuUrHPMiifn23sWfeiqR9wkYhm8EweDRL_vF-JSxMY,6642
 model_compression_toolkit/exporter/model_wrapper/__init__.py,sha256=7CF2zvpTrIEm8qnbuHnLZyTZkwBBxV24V8QA0oxGbh0,1187
 model_compression_toolkit/exporter/model_wrapper/fw_agnostic/__init__.py,sha256=pKAdbTCFM_2BrZXUtTIw0ouKotrWwUDF_hP3rPwCM2k,696
 model_compression_toolkit/exporter/model_wrapper/fw_agnostic/get_inferable_quantizers.py,sha256=Bd3QhAR__YC9Xmobd5qHv9ofh_rPn_eTFV0sXizcBnY,2297
@@ -528,7 +528,7 @@ model_compression_toolkit/xquant/pytorch/model_analyzer.py,sha256=b93o800yVB3Z-i
 model_compression_toolkit/xquant/pytorch/pytorch_report_utils.py,sha256=UVN_S9ULHBEldBpShCOt8-soT8YTQ5oE362y96qF_FA,3950
 model_compression_toolkit/xquant/pytorch/similarity_functions.py,sha256=CERxq5K8rqaiE-DlwhZBTUd9x69dtYJlkHOPLB54vm8,2354
 model_compression_toolkit/xquant/pytorch/tensorboard_utils.py,sha256=mkoEktLFFHtEKzzFRn_jCnxjhJolK12TZ5AQeDHzUO8,9767
-mct_nightly-2.3.0.20250521.607.dist-info/METADATA,sha256=J2U62R22BzYc8p77Ch7GUg96zt1YzVP2i_jN71fyJQA,25135
-mct_nightly-2.3.0.20250521.607.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-mct_nightly-2.3.0.20250521.607.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-2.3.0.20250521.607.dist-info/RECORD,,
+mct_nightly-2.3.0.20250523.601.dist-info/METADATA,sha256=G5J-yLvF68u133KBY6u5mACfZlFlONNxsM9ZoQZznUk,25135
+mct_nightly-2.3.0.20250523.601.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+mct_nightly-2.3.0.20250523.601.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-2.3.0.20250523.601.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.3.0.20250521.000607"
+__version__ = "2.3.0.20250523.000601"

model_compression_toolkit/exporter/model_exporter/pytorch/base_pytorch_exporter.py

@@ -19,8 +19,42 @@ import torch.nn
 
 from mct_quantizers import PytorchQuantizationWrapper
 from mct_quantizers.common.constants import LAYER, WEIGHTS_QUANTIZERS
+from model_compression_toolkit.logger import Logger
 from model_compression_toolkit.exporter.model_exporter.fw_agonstic.exporter import Exporter
 
+def find_and_assign_metadata_attr(model: torch.nn.Module, attr_name: str = 'metadata'):
+    """
+    Searches for a given attribute in the model and its submodules.
+    If found, assigns the first occurrence to the top-level model under the same attribute name.
+    Warn if the attribute is not found or found in multiple places.
+
+    Args:
+        model (torch.nn.Module): The model to search.
+        attr_name (str): The name of the attribute to look for. Default is 'metadata'.
+    """
+    found_attrs = []
+
+    def _search(m):
+        """Recursively search the model and its submodules for the attribute."""
+        if hasattr(m, attr_name):
+            found_attrs.append(getattr(m, attr_name))
+        for child in m.children():
+            _search(child)
+
+    _search(model)
+
+    if not found_attrs:
+        # Warn if the attribute was not found anywhere
+        Logger.warning(f"Attribute '{attr_name}' not found in the model or its submodules.")
+    else:
+        setattr(model, attr_name, found_attrs[0])
+
+        if len(found_attrs) > 1:
+            # Warn if the attribute was found in multiple places
+            Logger.warning(
+                f"Attribute '{attr_name}' found in {len(found_attrs)} places. "
+                f"Only the first one was assigned to 'model.metadata'.")
+
 
 def _set_quantized_weights_in_wrapper(layer:PytorchQuantizationWrapper):
     """

model_compression_toolkit/exporter/model_exporter/pytorch/pytorch_export_facade.py

@@ -30,6 +30,8 @@ if FOUND_TORCH:
     from model_compression_toolkit.exporter.model_exporter.pytorch.fakely_quant_onnx_pytorch_exporter import FakelyQuantONNXPyTorchExporter
     from model_compression_toolkit.exporter.model_exporter.pytorch.fakely_quant_torchscript_pytorch_exporter import FakelyQuantTorchScriptPyTorchExporter
     from model_compression_toolkit.exporter.model_wrapper.pytorch.validate_layer import is_pytorch_layer_exportable
+    from model_compression_toolkit.exporter.model_exporter.pytorch.base_pytorch_exporter import \
+        find_and_assign_metadata_attr
 
     if version.parse(torch.__version__) >= version.parse("2.4"):
         DEFAULT_ONNX_OPSET_VERSION = 20
@@ -65,6 +67,8 @@ if FOUND_TORCH:
             onnx_opset_version: ONNX opset version to use for exported ONNX model.
 
         """
+        # Ensure 'metadata' is available directly on the model, if present in submodules
+        find_and_assign_metadata_attr(model)
 
         if serialization_format == PytorchExportSerializationFormat.TORCHSCRIPT:
             if quantization_format in supported_serialization_quantization_export_dict[serialization_format]: