Skip to content

Commit 8932e2c

Browse files
authored
Update hf_ptq.py
Signed-off-by: Chenjie Luo <[email protected]>
1 parent 2d98b37 commit 8932e2c

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

examples/llm_ptq/hf_ptq.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -329,9 +329,7 @@ def main(args):
329329
model_type = get_model_type(model)
330330

331331
if model_type == "phi4mm":
332-
warnings.warn(
333-
"Please set the default input_mode to InputMode.LANGUAGE before quantizing."
334-
)
332+
warnings.warn("Please set the default input_mode to InputMode.LANGUAGE before quantizing.")
335333

336334
if args.sparsity_fmt != "dense":
337335
if args.batch_size == 0:

0 commit comments

Comments
 (0)