remove post calib hook

Signed-off-by: Kinjal Patel <[email protected]>

Author: kinjalpatel27

modelopt/torch/quantization/mode.py

@@ -208,8 +208,6 @@ def wrapped_calib_func(
     forward_loop and the relevant kwargs and are independent of the ModelOpt framework.
     So lets wrap them to be compatible with the ModelOpt convert entrypoint.
     """
-    from .plugins.custom import register_custom_post_calibration_plugins
-
     kwargs = config.model_dump()
     method = kwargs.pop("method")
     if method is not None and "awq" in method:
@@ -220,7 +218,6 @@ def wrapped_calib_func(
         # Call the function with forward_loop as a separate argument
         func(model, forward_loop=forward_loop, **kwargs)
 
-    register_custom_post_calibration_plugins(model)
     # Lets get the latest metadata for the quantizer states
     metadata = {}
     update_quantize_metadata(model, config, metadata)

modelopt/torch/quantization/plugins/custom.py

@@ -30,7 +30,6 @@
 
 CUSTOM_MODEL_PLUGINS = set()
 CUSTOM_POST_CONVERSION_PLUGINS = set()
-CUSTOM_POST_CALIBRATION_PLUGINS = set()
 
 
 # TODO: This is a temporary solution
@@ -47,12 +46,6 @@ def register_custom_post_conversion_plugins(model):
         callback(model)
 
 
-def register_custom_post_calibration_plugins(model):
-    """Registers custom modules as QUANT_MODULE after calibration."""
-    for callback in CUSTOM_POST_CALIBRATION_PLUGINS:
-        callback(model)
-
-
 class _QuantFunctionalMixin(QuantModule):
     """Mixin class for quantized functionals.