p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/llamafactory/hparams/generating_args.py
View File

@ -1,5 +1,5 @@
from dataclasses import asdict, dataclass, field from dataclasses import asdict, dataclass, field
from typing import Any, Dict from typing import Any, Dict, Optional
@dataclass @dataclass
@ -46,10 +46,9 @@ class GeneratingArguments:
default=1.0, default=1.0,
metadata={"help": "Exponential penalty to the length that is used with beam-based generation."}, 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, default=None,
metadata={ metadata={"help": "Default system message to use in chat completion."},
"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]: def to_dict(self) -> Dict[str, Any]: