remove full stop

pytorch · Jul 26, 2024 · 6cec214 · 6cec214
1 parent c98509a
commit 6cec214
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/torchao/prototype/low_bit_optim/subclass_4bit.py b/torchao/prototype/low_bit_optim/subclass_4bit.py
@@ -32,7 +32,7 @@ def __new__(cls, codes: Tensor, scale: Tensor, qmap: Tensor, signed: bool, shape
         )
 
     def __init__(self, codes: Tensor, scale: Tensor, qmap: Tensor, signed: bool, shape):
-        """Create quantized 4-bit optimizer state as proposed in https://arxiv.org/abs/2309.01507.
+        """Create quantized 4-bit optimizer state as proposed in https://arxiv.org/abs/2309.01507
 
         Args
             codes: quantized and packed 4-bit data stored as uint8.

diff --git a/torchao/prototype/low_bit_optim/subclass_8bit.py b/torchao/prototype/low_bit_optim/subclass_8bit.py
@@ -27,7 +27,7 @@ def __new__(cls, codes: Tensor, scale: Tensor, qmap: Tensor, signed: bool):
         )
 
     def __init__(self, codes: Tensor, scale: Tensor, qmap: Tensor, signed: bool):
-        """Create quantized 8-bit optimizer state as proposed in https://arxiv.org/abs/2110.02861.
+        """Create quantized 8-bit optimizer state as proposed in https://arxiv.org/abs/2110.02861
 
         Args
             codes: quantized 8-bit data stored as uint8. Has the same shape as the original float tensor.