remove print

Signed-off-by: Jennifer Chen <jennifchen@nvidia.com>

Author: jenchen13

modelopt/torch/quantization/plugins/megatron.py

@@ -594,7 +594,6 @@ def layer_sync_moe_local_experts_amax(self):
         amax_dict = {}
         for expert in self.local_experts:
             for name, module in expert.named_modules():
-                print(name, module)
                 if (
                     isinstance(module, TensorQuantizer)
                     and module.amax is not None

tests/gpu_megatron/torch/quantization/plugins/test_megatron.py

@@ -841,7 +841,6 @@ def _test_expert_model_parallel_amax_sync(
 
     # quantize the model
     model = mtq.quantize(model, config, forward)
-
     # Check initial sync status
     initial_sync, quantizer_type, rank_values = compare_amax_sync_across_expert_parallel(model)
     assert initial_sync, (