fix param type

Former-commit-id: 11b99f344416ade1cdac52e11ba7f36fcf689221
2023-12-21 17:33:01 +08:00
parent 79f301a2c6
commit 67f7034a21
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."}
    )