Skip to content

Commit 6cec214

Browse files
committed
remove full stop
1 parent c98509a commit 6cec214

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

torchao/prototype/low_bit_optim/subclass_4bit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def __new__(cls, codes: Tensor, scale: Tensor, qmap: Tensor, signed: bool, shape
3232
)
3333

3434
def __init__(self, codes: Tensor, scale: Tensor, qmap: Tensor, signed: bool, shape):
35-
"""Create quantized 4-bit optimizer state as proposed in https://arxiv.org/abs/2309.01507.
35+
"""Create quantized 4-bit optimizer state as proposed in https://arxiv.org/abs/2309.01507
3636
3737
Args
3838
codes: quantized and packed 4-bit data stored as uint8.

torchao/prototype/low_bit_optim/subclass_8bit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def __new__(cls, codes: Tensor, scale: Tensor, qmap: Tensor, signed: bool):
2727
)
2828

2929
def __init__(self, codes: Tensor, scale: Tensor, qmap: Tensor, signed: bool):
30-
"""Create quantized 8-bit optimizer state as proposed in https://arxiv.org/abs/2110.02861.
30+
"""Create quantized 8-bit optimizer state as proposed in https://arxiv.org/abs/2110.02861
3131
3232
Args
3333
codes: quantized 8-bit data stored as uint8. Has the same shape as the original float tensor.

0 commit comments

Comments
 (0)