Update generating_args.py

Former-commit-id: f32eefae3d
This commit is contained in:
hoshi-hiyouga
2024-05-07 00:28:16 +08:00
committed by GitHub
parent d65b2332cf
commit 4c91104471

View File

@@ -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: