address review feedbacks

Signed-off-by: realAsma <[email protected]>

Author: realAsma

modelopt/torch/quantization/model_calib.py

@@ -574,14 +574,9 @@ def forward(self, input, *args, **kwargs):
 
             disable_awq = False
             for tq in [self.input_quantizer, self.weight_quantizer]:
-                for attr in ["_pre_quant_scale", "_amax"]:
-                    if not tq.validate_attr(attr_name=attr):
-                        disable_awq = True
-                        warnings.warn(
-                            f"awq_lite: {attr} is not valid for {self.awq_lite.name}, skipping awq_lite"
-                        )
-                        break
-                if disable_awq:
+                if not tq.validate_attr(attr_name="_pre_quant_scale"):
+                    disable_awq = True
+                    warnings.warn(f"awq_lite: skipping awq_lite for valid for {self.awq_lite.name}")
                     break
 
             if disable_awq:

modelopt/torch/quantization/nn/modules/tensor_quantizer.py

@@ -514,7 +514,7 @@ def validate_attr(
         is_valid = torch.all(attr_value >= 0) and not torch.any(torch.isinf(attr_value))
         if is_valid:
             return True
-        name = f"{name} " if name else ""
+        name = f"{name}." if name else ""
         msg = f"{name}{attr_name} contains invalid values: {attr_value}"
         if warn_error:
             warnings.warn(msg)