diff --git a/torchao/quantization/linear_activation_quantized_tensor.py b/torchao/quantization/linear_activation_quantized_tensor.py index dfe1f62de7..b8070aff69 100644 --- a/torchao/quantization/linear_activation_quantized_tensor.py +++ b/torchao/quantization/linear_activation_quantized_tensor.py @@ -39,6 +39,9 @@ def __init__( self.original_weight_tensor = original_weight_tensor self.input_quant_func = input_quant_func + def __repr__(self): + return f"LinearActivationQuantizedTensor({self.original_weight_tensor}, {self.input_quant_func})" + def __tensor_flatten__(self): return ["original_weight_tensor"], [self.input_quant_func]