fix method name in error message

sony · Jun 26, 2024 · abfb19e · abfb19e
1 parent b5243eb
commit abfb19e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model_compression_toolkit/gptq/pytorch/quantization_facade.py b/model_compression_toolkit/gptq/pytorch/quantization_facade.py
@@ -167,7 +167,7 @@ def pytorch_gradient_post_training_quantization(model: Module,
         if core_config.mixed_precision_enable:
             if not isinstance(core_config.mixed_precision_config, MixedPrecisionQuantizationConfig):
                 Logger.critical("Given quantization config for mixed-precision is not of type 'MixedPrecisionQuantizationConfig'. "
-                                "Ensure usage of the correct API for 'keras_post_training_quantization' "
+                                "Ensure usage of the correct API for 'pytorch_gradient_post_training_quantization' "
                                 "or provide a valid mixed-precision configuration.")  # pragma: no cover
 
         tb_w = init_tensorboard_writer(DEFAULT_PYTORCH_INFO)