Update generating_args.py

2024-05-07 00:28:16 +08:00 · 2024-05-07 00:28:16 +08:00 · f32eefae3d
parent 7ae7ae64f0
commit f32eefae3d
1 changed files with 1 additions and 0 deletions
--- a/src/llmtuner/hparams/generating_args.py
+++ b/src/llmtuner/hparams/generating_args.py
@ -46,6 +46,7 @@ class GeneratingArguments:
        default=1.0,
        metadata={"help": "Exponential penalty to the length that is used with beam-based generation."},
    )
+
    def to_dict(self) -> Dict[str, Any]:
        args = asdict(self)
        if args.get("max_new_tokens", -1) > 0: