Skip to content

Commit f48a948

Browse files
fix lint on main branch
Signed-off-by: cliu-us <[email protected]>
1 parent dceb63b commit f48a948

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

fms_mo/run_quant.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,6 @@ def run_gptq(model_args, data_args, opt_args, gptq_args):
155155
v2_memory_device="cpu",
156156
)
157157

158-
159158
# Add custom model_type mapping to gptqmodel LUT so GPTQModel can recognize them.
160159
for mtype, cls in custom_gptq_classes.items():
161160
if mtype in MODEL_MAP:

fms_mo/training_args.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -139,8 +139,8 @@ class OptArguments(TypeChecker):
139139

140140
quant_method: str = field(
141141
metadata={
142-
"choices": ["gptq", "gptqv2", "fp8", "dq"],
143-
"help": "Quantization technique"
142+
"choices": ["gptq", "gptqv2", "fp8", "dq"],
143+
"help": "Quantization technique",
144144
}
145145
)
146146
output_dir: str = field(
@@ -229,7 +229,6 @@ class GPTQArguments(TypeChecker):
229229
cache_examples_on_gpu: bool = True
230230

231231

232-
233232
@dataclass
234233
class FP8Arguments(TypeChecker):
235234
"""Dataclass for FP8 related arguments that will be used by llm-compressor."""

0 commit comments

Comments
 (0)