From a710d97748bebb6aa702df9a41dcab457d51184e Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Mon, 20 May 2024 00:29:31 +0800 Subject: [PATCH] Update generating_args.py Former-commit-id: a1fa7aa63b9b3fade3de6bd27395c1b94068b6d2 --- src/llamafactory/hparams/generating_args.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/llamafactory/hparams/generating_args.py b/src/llamafactory/hparams/generating_args.py index 17669a51..0ee17d1a 100644 --- a/src/llamafactory/hparams/generating_args.py +++ b/src/llamafactory/hparams/generating_args.py @@ -1,5 +1,5 @@ from dataclasses import asdict, dataclass, field -from typing import Any, Dict +from typing import Any, Dict, Optional @dataclass @@ -46,10 +46,9 @@ class GeneratingArguments: default=1.0, metadata={"help": "Exponential penalty to the length that is used with beam-based generation."}, ) - system_message: str = field( + default_system: Optional[str] = field( default=None, - metadata={ - "help": "System message is a message that the developer wrote to tell the bot how to interpret the conversation"}, + metadata={"help": "Default system message to use in chat completion."}, ) def to_dict(self) -> Dict[str, Any]: