Skip to content

Commit 4bf35d0

Browse files
Jack-Khuufacebook-github-bot
authored andcommitted
Change the default argument for Calibration Tasks for GPTQ
Summary: GPTQ is the sole user of the `Calibration_Tasks` param. Previously the default value was [], which is not something `Int8DynActInt4WeightGPTQQuantizer` handles gracefully. This made the field effectively always required for GPTQ. By setting it to None, we allow quantize to resolve to ["wikitext"] and save a little bit of typing Differential Revision: D55458866
1 parent ef56bf1 commit 4bf35d0

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

examples/models/llama2/export_llama_lib.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -341,8 +341,7 @@ def build_args_parser() -> argparse.ArgumentParser:
341341
parser.add_argument(
342342
"--calibration_tasks",
343343
nargs="+",
344-
type=str,
345-
default=[],
344+
default=None,
346345
help="Tasks for GPTQ calibration",
347346
)
348347
parser.add_argument(

0 commit comments

Comments
 (0)