feat: cli chat support system_message

Former-commit-id: e3982bff596d01992733687a580c4f41c558061c
This commit is contained in:
ycjcl868
2024-05-19 23:17:46 +08:00
parent 516d83c946
commit 77a089c35c
2 changed files with 7 additions and 0 deletions

View File

@@ -46,6 +46,11 @@ class GeneratingArguments:
default=1.0,
metadata={"help": "Exponential penalty to the length that is used with beam-based generation."},
)
system_message: str = field(
default=None,
metadata={
"help": "System message is a message that the developer wrote to tell the bot how to interpret the conversation"},
)
def to_dict(self) -> Dict[str, Any]:
args = asdict(self)