File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
src/llmcompressor/modifiers/smoothquant Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 44import torch
55from compressed_tensors .utils import align_module_device
66from loguru import logger
7- from pydantic import Field , ConfigDict
7+ from pydantic import ConfigDict , Field
88from torch .nn import Module
99
1010from 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 :
You can’t perform that action at this time.
0 commit comments