Minor bug fix

Signed-off-by: Suguna Velury <[email protected]>

Author: sugunav14

modelopt/torch/export/quant_utils.py

@@ -608,6 +608,8 @@ def process_layer_quant_config(layer_config_dict):
         # Get the corresponding AWQ block size
         block_size_value = layer_config_dict.get(awq_key, 0)
 
+        # print(f"DEBUG LOG: Processing layer {k} with quantization {v}, block size {block_size_value}")
+
         if v == "fp8":
             layer_config = {"quant_algo": "FP8"}
         elif v == "fp8_pc_pt":
@@ -1079,9 +1081,11 @@ def get_quant_config(named_modules: nn.Module | dict[str, nn.Module]) -> dict[st
             if block_size == 0:
                 block_size = get_weight_block_size(module)
 
-            # Construct per layer config dictionary
-            layer_config_dict[name + ".quantization"] = quantization_format
-            layer_config_dict[name + ".awq_block_size"] = block_size
+            # Handles case if default weight quantizer is not enabled or is None
+            if block_size != 0:
+                # Construct per layer config dictionary
+                layer_config_dict[name + ".quantization"] = quantization_format
+                layer_config_dict[name + ".awq_block_size"] = block_size
 
         # Find kv cache quant format
         if (