fix param type

2023-12-21 17:33:01 +08:00 · 2023-12-21 17:33:01 +08:00 · ba69378841
parent 083355fc05
commit ba69378841
1 changed files with 1 additions and 1 deletions
--- a/src/llmtuner/hparams/finetuning_args.py
+++ b/src/llmtuner/hparams/finetuning_args.py
@ -157,7 +157,7 @@ class ExportArguments:
        default=128,
        metadata={"help": "The number of samples used for quantization."}
    )
-    export_quantization_maxlen: Optional[str] = field(
+    export_quantization_maxlen: Optional[int] = field(
        default=1024,
        metadata={"help": "The maximum length of the model inputs used for quantization."}
    )