Clean up code, more

jingyu-ml · jingyu-ml · commit 7326e3859e0e · 2025-09-18T21:41:16.000Z
Signed-off-by: Jingyu Xin &lt;jingyux@nvidia.com&gt;
diff --git a/modelopt/torch/quantization/conversion.py b/modelopt/torch/quantization/conversion.py
@@ -176,7 +176,7 @@ def replace_quant_module(model: nn.Module, version=None, registry=QuantModuleReg
     """Recursively replace the module with quantized module."""
     from .plugins.custom import (
         register_custom_model_plugins_on_the_fly,
-        register_custom_post_conversion_plugins, ## not needed for lora megatron
+        register_custom_post_conversion_plugins,
     )
 
     assert not is_quantized(model), "Model must not be quantized!"

Original file line number	Diff line number	Diff line change
`@@ -176,7 +176,7 @@ def replace_quant_module(model: nn.Module, version=None, registry=QuantModuleReg`
`176`	`176`	`"""Recursively replace the module with quantized module."""`
`177`	`177`	`from .plugins.custom import (`
`178`	`178`	`register_custom_model_plugins_on_the_fly,`
`179`		`- register_custom_post_conversion_plugins, ## not needed for lora megatron`
	`179`	`+ register_custom_post_conversion_plugins,`
`180`	`180`	`)`
`181`	`181`
`182`	`182`	`assert not is_quantized(model), "Model must not be quantized!"`