Skip to content
This repository was archived by the owner on Oct 25, 2024. It is now read-only.

Commit fbbd653

Browse files
authored
update LLM recipes (#1420)
Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com>
1 parent c423deb commit fbbd653

File tree

3 files changed

+715
-21
lines changed

3 files changed

+715
-21
lines changed

examples/huggingface/pytorch/code-generation/quantization/run_generation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from intel_extension_for_transformers.transformers import (
2525
AutoModelForCausalLM,
2626
)
27+
from intel_extension_for_transformers.transformers.utils import str2bool
2728

2829
parser = argparse.ArgumentParser()
2930

@@ -63,7 +64,6 @@
6364
parser.add_argument("--mixed_precision", action="store_true")
6465
# ============SmoothQuant configs==============
6566
parser.add_argument("--sq", action="store_true")
66-
parser.add_argument("--calib_iters", default=100, type=int, help="Calibration iters.")
6767
parser.add_argument(
6868
"--calib_padding", action="store_true", help="Calibration dataset do padding."
6969
)

0 commit comments

Comments
 (0)