Skip to content

Commit 425e72d

Browse files
committed
remove comments
1 parent 73e1a06 commit 425e72d

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

torchao/quantization/quantize_/workflows/int4/int4_marlin_sparse_tensor.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131

3232

3333
class Int4MarlinSparseTensor(TorchAOBaseTensor):
34-
tensor_data_names = ["qdata", "scale", "zero_point", "meta"] # meta is a tensor
34+
tensor_data_names = ["qdata", "scale", "zero_point", "meta"]
3535
tensor_attribute_names = ["block_size", "num_bits", "shape"]
3636

3737
def __new__(cls, qdata, scale, zero_point, meta, block_size, num_bits, shape):
@@ -41,9 +41,7 @@ def __new__(cls, qdata, scale, zero_point, meta, block_size, num_bits, shape):
4141
kwargs["requires_grad"] = False
4242
return torch.Tensor._make_wrapper_subclass(cls, shape, **kwargs) # type: ignore[attr-defined]
4343

44-
def __init__(
45-
self, qdata, scale, zero_point, meta, block_size, num_bits, shape
46-
): # args need to match lines 37 and 38
44+
def __init__(self, qdata, scale, zero_point, meta, block_size, num_bits, shape):
4745
self.qdata = qdata
4846
self.scale = scale
4947
self.zero_point = zero_point

0 commit comments

Comments
 (0)