Skip to content

Commit a4abb3d

Browse files
stylefixes
Signed-off-by: Brian Dellabetta <bdellabe@redhat.com>
1 parent 31ac8e9 commit a4abb3d

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/llmcompressor/modifiers/transform/spinquant/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ class SpinquantRotation(str, Enum):
2828

2929
class SpinQuantModifier(Modifier, use_enum_values=True):
3030
"""
31-
Implements the transforms according to
32-
[SpinQuant: LLM quantization with learned rotations](https://arxiv.org/abs/2405.16406) # noqa: E501
31+
Implements the transforms according to "SpinQuant: LLM quantization
32+
with learned rotations" (https://arxiv.org/abs/2405.16406)
3333
3434
Transforms (rotations) are extra layers added to a model which reduce the accuracy
3535
loss induced by quantization. This is achived through "rotating" weights and

src/llmcompressor/modifiers/transform/spinquant/norm_mappings.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ class NormMapping(BaseModel):
1818
:param linears: list of names or regexes of Linear layers that
1919
receive input from norm.
2020
"""
21+
2122
norm: str
2223
linears: List[str]
2324

0 commit comments

Comments
 (0)