Update generating_args.py

This commit is contained in:
hoshi-hiyouga 2024-05-20 00:29:31 +08:00 committed by GitHub
parent 896c656185
commit a1fa7aa63b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions

View File

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