minor fixes

sony · Jan 1, 2025 · b89ca55 · b89ca55
1 parent 1c909b9
commit b89ca55
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/model_compression_toolkit/gptq/pytorch/quantization_facade.py b/model_compression_toolkit/gptq/pytorch/quantization_facade.py
@@ -144,7 +144,7 @@ def pytorch_gradient_post_training_quantization(model: Module,
                                                     core_config: CoreConfig = CoreConfig(),
                                                     gptq_config: GradientPTQConfig = None,
                                                     gptq_representative_data_gen: Callable = None,
-                                                    target_platform_capabilities: TargetPlatformModel = None):
+                                                    target_platform_capabilities: TargetPlatformModel = DEFAULT_PYTORCH_TPC):
         """
         Quantize a trained Pytorch module using post-training quantization.
         By default, the module is quantized using a symmetric constraint quantization thresholds

diff --git a/model_compression_toolkit/target_platform_capabilities/target_platform/get_tpc_model.py b/model_compression_toolkit/target_platform_capabilities/target_platform/get_tpc_model.py
diff --git a/...oolkit/target_platform_capabilities/target_platform/targetplatform2framework/attach2fw.py b/...oolkit/target_platform_capabilities/target_platform/targetplatform2framework/attach2fw.py
@@ -37,7 +37,6 @@ def attach(self, tpc_model: TargetPlatformModel,
         """
 
         tpc = TargetPlatformCapabilities(tpc_model)
-        # tpc_model_opsets = [opset.name for opset in tpc_model.operator_set if isinstance(opset, OperatorsSet)]
         custom_opset2layer = custom_opset2layer if custom_opset2layer is not None else {}
 
         with tpc: