Skip to content

Commit 92c9dee

Browse files
committed
style
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent d311fb3 commit 92c9dee

File tree

1 file changed

+4
-2
lines changed
  • src/llmcompressor/modifiers/smoothquant

1 file changed

+4
-2
lines changed

src/llmcompressor/modifiers/smoothquant/base.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import torch
55
from compressed_tensors.utils import align_module_device
66
from loguru import logger
7-
from pydantic import Field, ConfigDict
7+
from pydantic import ConfigDict, Field
88
from torch.nn import Module
99

1010
from llmcompressor.core import Event, EventType, State
@@ -104,7 +104,9 @@ class SmoothQuantModifier(Modifier):
104104
num_calibration_steps: Optional[int] = None
105105
calibration_function: Optional[Callable] = None
106106

107-
resolved_mappings_: Optional[List[SmoothQuantMapping]] = Field(default=None, repr=False)
107+
resolved_mappings_: Optional[List[SmoothQuantMapping]] = Field(
108+
default=None, repr=False
109+
)
108110
scales_: Optional[Dict] = Field(default=None, repr=False)
109111

110112
def on_initialize(self, state: State, **kwargs) -> bool:

0 commit comments

Comments
 (0)